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

sgemm_wg: save args and input bin

Browse Source
This commit is contained in:
Hansung Kim
2024-02-12 20:48:23 -08:00
parent f586ab28df
commit 6a1a506b64
4 changed files with 46 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/regression/sgemm_wg/kernel.cpp
View File

@@ -4,15 +4,23 @@
#include "common.h"
void kernel_body(int task_id, kernel_arg_t* __UNIFORM__ arg) {
uint32_t num_points = arg->num_points;
float* src_ptr = (float*)arg->src_addr;
float* dst_ptr = (float*)arg->dst_addr;
uint32_t num_points = arg->num_points;
float *src_ptr = (float *)arg->src_addr;
float *dst_ptr = (float *)arg->dst_addr;
dst_ptr[task_id] = 2 * src_ptr[task_id];
float *local_a = (float *)DEV_SMEM_START_ADDR;
local_a[num_points - 1 - task_id] = 2 * src_ptr[num_points - 1 - task_id];
// local_a[task_id] = 2 * src_ptr[task_id];
vx_barrier(0, vx_num_warps());
dst_ptr[task_id] = local_a[task_id];
}
int main() {
kernel_arg_t* arg = (kernel_arg_t*)KERNEL_ARG_DEV_MEM_ADDR;
vx_spawn_tasks(arg->num_points, (vx_spawn_tasks_cb)kernel_body, arg);
int threads_per_core = vx_num_warps() * vx_num_threads();
vx_spawn_tasks(threads_per_core, (vx_spawn_tasks_cb)kernel_body, arg);
return 0;
}