ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-17 09:08:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

add lds barrier

Browse Source
This commit is contained in:
joye
2025-06-11 12:44:03 +08:00
parent 4c85e10281
commit 09a2e029b9
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_data_multiple_d_xdl_cshuffle_v1.hpp
View File

@@ -511,6 +511,7 @@ __global__ void kernel_grouped_conv_bwd_data_optimized_v2(const ABDataType* __re
#pragma unroll
for(int batch_id = 0; batch_id < BatchPerBlock; batch_id += WaveNum)
{
block_sync_lds();
for(int i = tid; i < InLoopNum; i += BlockSize)
{
int group_id = i % GroupPerBlockInFP4;