ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-08 15:30:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add multiple Ds instances

Browse Source
This commit is contained in:
Enrico Degregori
2025-08-06 13:00:33 +00:00
parent e6b7d5ed65
commit c71f2f25eb
7 changed files with 171 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_multiple_d_wmma_cshuffle_v3.hpp
View File

@@ -273,8 +273,8 @@ struct DeviceGroupedConvBwdWeightMultipleD_Wmma_CShuffleV3
using CGridDesc_M_N = remove_cvref_t<decltype(ABCGridDescs{}[I2])>;
using GridwiseGemm = GridwiseGemm_wmma_cshuffle_v3<
tensor_layout::gemm::RowMajor,
tensor_layout::gemm::ColumnMajor,
tensor_layout::gemm::RowMajor,
Tuple<>,
tensor_layout::gemm::RowMajor,
Tuple<ADataType>,

2
include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_two_stage_wmma_cshuffle_v3.hpp
View File

@@ -323,8 +323,8 @@ struct DeviceGroupedConvBwdWeightTwoStage_Wmma_CShuffleV3
remove_cvref_t<decltype(GetElementwiseCGridDesc<NDimSpatial>())>;
using GridwiseGemm = GridwiseGemm_wmma_cshuffle_v3<
tensor_layout::gemm::RowMajor,
tensor_layout::gemm::ColumnMajor,
tensor_layout::gemm::RowMajor,
Tuple<>,
tensor_layout::gemm::RowMajor,
Tuple<ADataType>,