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

Change constant to variable

Browse Source
This commit is contained in:
mtgu0705
2024-12-30 19:47:30 +08:00
parent 1867ffa36d
commit 5e91d8444f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/ck/tensor_operation/gpu/block/blockwise_gemm_pipeline_xdlops_v3_ab_scale.hpp
View File

@@ -339,7 +339,7 @@ struct BlockwiseGemmXdlops_pipeline_v3_ab_scale<BlockGemmPipelineScheduler::Intr
b_blockwise_copy.MoveSrcSliceWindow(b_grid_desc, b_block_copy_step);
static_for<0, MRepeat, 1>{}([&](auto m0) {
static_for<0, Number<16>{}, 1>{}([&](auto i0) {
static_for<0, Number<xdlops_gemm.GetRegSizePerXdlops()>{}, 1>{}([&](auto i0) {
a_scale_thread_copy.Run(a_scale_grid_desc,
a_scale_grid_buf,
a_scale_thread_desc,
@@ -479,7 +479,7 @@ struct BlockwiseGemmXdlops_pipeline_v3_ab_scale<BlockGemmPipelineScheduler::Intr
});
});
static_for<0, MRepeat, 1>{}([&](auto m0) {
static_for<0, Number<16>{}, 1>{}([&](auto i0) {
static_for<0, Number<xdlops_gemm.GetRegSizePerXdlops()>{}, 1>{}([&](auto i0) {
a_scale_thread_copy.Run(a_scale_grid_desc,
a_scale_grid_buf,
a_scale_thread_desc,