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

sgemm_tcore: Constify smem pointer for wmma_load

Browse Source
This commit is contained in:
Hansung Kim
2024-06-11 22:49:59 -07:00
parent f73029889b
commit 635da96154
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/regression/sgemm_tcore/kernel.warpspecial.cpp
View File

@@ -381,8 +381,8 @@ inline void thread_block_gemm(kernel_arg_t *__UNIFORM__ arg,
int32_t k_index = 0;
#pragma GCC unroll 1
for (uint32_t k = 0; k < (dim_k); k += BK) {
volatile float *local_a_consume;
volatile float *local_b_consume;
const volatile float *local_a_consume;
const volatile float *local_b_consume;
if constexpr (DOUBLE_BUFFER) {
// local_a_consume = (k_index % 2) ? local_a_buf : local_a;
// local_b_consume = (k_index % 2) ? local_b_buf : local_b;