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

sgemm_tg: Fix device address to use ELF operands

Browse Source
This commit is contained in:
Hansung Kim
2024-05-13 23:09:57 -07:00
parent 9d2b533d5c
commit 5de8e7c33a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -169,8 +169,8 @@ int main(int argc, char *argv[]) {
// 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_a = 0xa0000000;
kernel_arg.addr_b = 0x28000; kernel_arg.addr_b = 0xa1000000;
kernel_arg.addr_c = 0xc0000000; kernel_arg.addr_c = 0xc0000000;
kernel_arg.dim_m = dim_m; kernel_arg.dim_m = dim_m;