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

Add 2GB limitation for grouped conv bwd weight (#3054)

Browse Source
This commit is contained in:
Bartłomiej Kocot
2025-11-01 14:16:45 +01:00
committed by GitHub
parent 1fbb47ad30
commit ab1a8356b6
4 changed files with 33 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3_multi_d.hpp
View File

@@ -1359,6 +1359,14 @@ struct GridwiseGemmMultiD_xdl_cshuffle_v3
}
}
constexpr long_index_t TwoGB = (long_index_t{1} << 31);
if(!(karg.M * karg.K * sizeof(ADataType) <= TwoGB &&
karg.N * karg.K * sizeof(BDataType) <= TwoGB &&
karg.M * karg.N * sizeof(CDataType) <= TwoGB))
{
return false;
}
// TODO: also check validity of all components (blockwise-copy, threadwise-copy, etc)
return true;
}

9
include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_bwd_weight.hpp
View File

@@ -581,6 +581,15 @@ struct GridwiseGemm_bk0mk1_bk0nk1_mn_xdlops_bwd_weight
return false;
}
constexpr long_index_t TwoGB = (long_index_t{1} << 31);
if(!(a_b_k0_m_k1_grid_desc.GetElementSpaceSize() * sizeof(FloatA) <= TwoGB &&
b_b_k0_n_k1_grid_desc.GetElementSpaceSize() * sizeof(FloatB) <= TwoGB &&
c_m_n_grid_desc.GetElementSpaceSize() * sizeof(FloatC) <= TwoGB))
{
return false;
}
// TODO: also check validity of all components (blockwise-copy, threadwise-copy, etc)
return true;
}