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

sgemm_tcore: Hardcode CISC spadQuartile addresses

Browse Source
This commit is contained in:
Hansung Kim
2024-10-28 12:49:22 -07:00
parent e55c8b480e
commit d0421426be
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -91,9 +91,9 @@ void kernel_body(int task_id, kernel_arg_t *__UNIFORM__ arg) {
/*write_to_gmem=*/true,
/*smem_a_offset=*/0,
#ifdef GEMMINI_DMA
/*smem_a_dbuf_offset=*/1 * 128 * 128 * sizeof(float_type),
/*smem_b_offset=*/2 * 128 * 128 * sizeof(float_type),
/*smem_b_dbuf_offset=*/3 * 128 * 128 * sizeof(float_type)
/*smem_a_dbuf_offset=*/1 * 128 * 128 * 2/*fp16*/,
/*smem_b_offset=*/2 * 128 * 128 * 2/*fp16*/,
/*smem_b_dbuf_offset=*/3 * 128 * 128 * 2/*fp16*/
// FIXME: above offsets are hardcoded to agree with CISC
// spadQuartile
#else