Skip to content

Commit 7d685e7

Browse files
committed
Fix for gemm_bias_add_reduce flavour
1 parent 582029c commit 7d685e7

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

include/ck/tensor_operation/gpu/device/impl/device_gemm_bias_add_reduce_wmma_cshuffle_v3.hpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,10 @@ struct DeviceGemmBiasAddReduce_Wmma_CShuffleV3
188188
ComputeTypeA,
189189
ComputeTypeB,
190190
PermuteA,
191-
PermuteB>;
191+
PermuteB,
192+
false, // IsBPreShuffled
193+
false, // ForceThreadTileTransfer
194+
true>; // IsFusedKernel
192195

193196
using ReduceTrait = ReduceTrait_<ReduceAccDataType,
194197
ReducePtrsGlobal,

0 commit comments

Comments
 (0)