Skip to content

Commit aa46039

Browse files
zjing14Jing Zhang
andauthored
Fixed contraction issues (ROCm#960)
* add missing ComputeType * fixed * Update cmake-ck-dev.sh --------- Co-authored-by: Jing Zhang <jizha@amd.com>
1 parent f477fca commit aa46039

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -291,6 +291,7 @@ int main(int argc, char* argv[])
291291
BDataType,
292292
CShuffleDataType,
293293
AccDataType,
294+
A0DataType,
294295
PassThrough,
295296
BElementOp>;
296297

include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_abd_xdl_cshuffle.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -657,6 +657,7 @@ struct GridwiseGemmMultipleABD_xdl_cshuffle
657657
auto blockwise_gemm = BlockwiseGemmXdlops_k0mk1_k0nk1_m0n0m1n1m2m3m4n2_Selector<
658658
BlockSize,
659659
ComputeDataType,
660+
ComputeDataType,
660661
AccDataType,
661662
decltype(a_block_desc_ak0_m_ak1),
662663
decltype(b_block_desc_bk0_n_bk1),

0 commit comments

Comments
 (0)