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

fix n pos in scale

Browse Source
This commit is contained in:
qin letao
2025-04-02 07:36:46 +00:00
parent 916b760ed6
commit c1aefb42bf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3_b_scale.hpp
View File

@@ -1464,7 +1464,7 @@ struct GridwiseGemm_xdl_cshuffle_v3
1,
false>(
b_scale_grid_desc_bn_ak,
make_multi_index(block_n_id * NPerBlock / ScaleBlockN + b_thread_offset_n,
make_multi_index((block_n_id * NPerBlock + b_thread_offset_n) / ScaleBlockN,
b_thread_offset_k / ScaleBlockK));
constexpr auto b_scale_thread_slice_copy_step =