wu-arch/vortex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

relu test added

Browse Source
This commit is contained in:
Nico Castaneda
2023-10-06 13:20:31 -07:00
parent d69a64c32c
commit 8296e6be0f
11 changed files with 2434 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/regression/relu/kernel.c Normal file
View File

@@ -0,0 +1,26 @@
#include <stdint.h>
#include <vx_intrinsics.h>
#include <vx_spawn.h>
#include "common.h"
void kernel_body(int __DIVERGENT__ task_id, kernel_arg_t* arg) {
uint32_t num_points = arg->num_points;
uint32_t points_per_core = num_points / vx_num_warps();
int tid = vx_thread_lid();
int32_t* src_ptr = (int32_t*)arg->src_addr;
int32_t* dst_ptr = (int32_t*)arg->dst_addr;
int32_t ref_value = src_ptr[task_id];
int ref_negative = ref_value < 0;
if (ref_negative) {
ref_value = 0;
}
dst_ptr[task_id] = ref_value;
}
void main() {
kernel_arg_t* arg = (kernel_arg_t*)KERNEL_ARG_DEV_MEM_ADDR;
int num_warps = vx_num_warps();
vx_spawn_tasks(arg->num_points, (vx_spawn_tasks_cb)kernel_body, arg);
}