ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-05 20:55:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
Gino Lu
2025-09-14 23:40:28 -05:00
committed by mtgu0705
parent e92e67b8dd
commit 22586c3135
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
include/ck_tile/ops/gemm/warp/warp_gemm_impl.hpp
View File

@@ -92,11 +92,11 @@ struct WarpGemmImpl
c.get_thread_buffer().template set_as<CVec>(I0, c_vec);
}
template <typename CTensor,
template <index_t opselA,
index_t opselB,
typename CTensor,
typename ATensor,
typename BTensor,
index_t opselA,
index_t opselB,
bool post_nop_ = false>
CK_TILE_DEVICE void operator()(CTensor& c,
const ATensor& a,
@@ -150,7 +150,7 @@ struct WarpGemmImpl
return c;
}
template <typename ATensor, typename BTensor, index_t opselA, index_t opselB>
template <index_t opselA, index_t opselB, typename ATensor, typename BTensor>
CK_TILE_DEVICE auto operator()(const ATensor& a,
const BTensor& b,
const int32_t& a_scale,