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

Hardcode chipyard device addresses

Browse Source
This commit is contained in:
Hansung Kim
2024-05-07 16:30:30 -07:00
parent b4c812f9f8
commit 7775830814
2 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/regression/sgemm_tcore/kernel.cpp
View File

@@ -282,4 +282,4 @@ int main() {
// we can thus treat 1 through NW as a single threadblock for the purposes of the optimization. // we can thus treat 1 through NW as a single threadblock for the purposes of the optimization.
vx_spawn_tasks_contiguous(grid_size, (vx_spawn_tasks_cb)kernel_body, arg); vx_spawn_tasks_contiguous(grid_size, (vx_spawn_tasks_cb)kernel_body, arg);
return 0; return 0;
} }

17
tests/regression/sgemm_tcore/main.cpp
View File

@@ -58,9 +58,9 @@ static void parse_args(int argc, char **argv) {
void cleanup() { void cleanup() {
if (device) { if (device) {
vx_mem_free(device, kernel_arg.addr_a); // vx_mem_free(device, kernel_arg.addr_a);
vx_mem_free(device, kernel_arg.addr_b); // vx_mem_free(device, kernel_arg.addr_b);
vx_mem_free(device, kernel_arg.addr_c); // vx_mem_free(device, kernel_arg.addr_c);
vx_dev_close(device); vx_dev_close(device);
} }
} }
@@ -166,9 +166,12 @@ int main(int argc, char *argv[]) {
// allocate device memory // allocate device memory
std::cout << "allocate device memory" << std::endl; std::cout << "allocate device memory" << std::endl;
RT_CHECK(vx_mem_alloc(device, src_a_buf_size, VX_MEM_TYPE_GLOBAL, &kernel_arg.addr_a)); // RT_CHECK(vx_mem_alloc(device, src_a_buf_size, VX_MEM_TYPE_GLOBAL, &kernel_arg.addr_a));
RT_CHECK(vx_mem_alloc(device, src_b_buf_size, VX_MEM_TYPE_GLOBAL, &kernel_arg.addr_b)); // RT_CHECK(vx_mem_alloc(device, src_b_buf_size, VX_MEM_TYPE_GLOBAL, &kernel_arg.addr_b));
RT_CHECK(vx_mem_alloc(device, dst_buf_size, VX_MEM_TYPE_GLOBAL, &kernel_arg.addr_c)); // RT_CHECK(vx_mem_alloc(device, dst_buf_size, VX_MEM_TYPE_GLOBAL, &kernel_arg.addr_c));
kernel_arg.addr_a = 0x20000;
kernel_arg.addr_b = 0x28000;
kernel_arg.addr_c = 0xc0000000;
kernel_arg.dim_m = dim_m; kernel_arg.dim_m = dim_m;
kernel_arg.dim_n = dim_n; kernel_arg.dim_n = dim_n;
@@ -267,4 +270,4 @@ int main(int argc, char *argv[]) {
cleanup(); cleanup();
return 0; return 0;
} }