Skip to content

Commit e38a84d

Browse files
committed
fix clang format
1 parent eb29023 commit e38a84d

File tree

2 files changed

+36
-35
lines changed

2 files changed

+36
-35
lines changed

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

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -406,11 +406,12 @@ struct ABTransferThreadTiles
406406
// KPack / KRow / K1 - MNRepeat - K0 / KRow - MNWaves - KRow - MNPerWmma - K1
407407
return transform_tensor_descriptor(
408408
BlockDesc{},
409-
make_tuple(make_unmerge_transform(make_tuple(
410-
Number<ABK0 / (KPack / ABK1)>{}, KRow, Number<KPack / KRow / ABK1>{})),
411-
make_unmerge_transform(make_tuple(
412-
Number<MNRepeat>{}, Number<MNWaves>{}, Number<MNPerWmma>{})),
413-
make_pass_through_transform(Number<ABK1>{})),
409+
make_tuple(
410+
make_unmerge_transform(make_tuple(
411+
Number<ABK0 / (KPack / ABK1)>{}, KRow, Number<KPack / KRow / ABK1>{})),
412+
make_unmerge_transform(
413+
make_tuple(Number<MNRepeat>{}, Number<MNWaves>{}, Number<MNPerWmma>{})),
414+
make_pass_through_transform(Number<ABK1>{})),
414415
make_tuple(Sequence<0>{}, Sequence<1>{}, Sequence<2>{}),
415416
make_tuple(Sequence<2, 4, 0>{}, Sequence<1, 3, 5>{}, Sequence<6>{}));
416417
}
@@ -439,13 +440,14 @@ struct ABTransferThreadTiles
439440

440441
return transform_tensor_descriptor(
441442
desc1,
442-
make_tuple(make_pass_through_transform(Number<KPack / KPerWmmaBlk / KRow>{}),
443-
make_pass_through_transform(Number<MNRepeat>{}),
444-
make_merge_transform(make_tuple(Number<ABK0>{}, Number<ABK1 / KPack>{})),
445-
make_pass_through_transform(Number<MNWaves>{}),
446-
make_pass_through_transform(Number<KRow>{}),
447-
make_pass_through_transform(Number<MNPerWmma>{}),
448-
make_pass_through_transform(Number<KPerWmmaBlk>{})),
443+
make_tuple(
444+
make_pass_through_transform(Number<KPack / KPerWmmaBlk / KRow>{}),
445+
make_pass_through_transform(Number<MNRepeat>{}),
446+
make_merge_transform(make_tuple(Number<ABK0>{}, Number<ABK1 / KPack>{})),
447+
make_pass_through_transform(Number<MNWaves>{}),
448+
make_pass_through_transform(Number<KRow>{}),
449+
make_pass_through_transform(Number<MNPerWmma>{}),
450+
make_pass_through_transform(Number<KPerWmmaBlk>{})),
449451
make_tuple(Sequence<0>{},
450452
Sequence<1>{},
451453
Sequence<2, 3>{},

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

Lines changed: 22 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -559,29 +559,28 @@ struct GridwiseGemm_wmma_cshuffle_v3_base
559559
Number<NumDTensor>{});
560560
}
561561

562-
using BlockwiseGemmPipe = remove_cvref_t<
563-
decltype(BlockGemmPipeline_Selector<
564-
BlkGemmPipelineVer,
565-
BlkGemmPipeSched,
566-
BlockSize,
567-
LDSTypeA,
568-
LDSTypeB,
569-
ComputeTypeA,
570-
ComputeTypeB,
571-
AccDataType,
572-
decltype(MakeAWmmaTileDescriptor()),
573-
decltype(MakeBWmmaTileDescriptor()),
574-
ABlockTransferSrcScalarPerVector,
575-
BBlockTransferSrcScalarPerVector,
576-
MPerBlock,
577-
NPerBlock,
578-
KPerBlock,
579-
MPerWmma,
580-
NPerWmma,
581-
MRepeat,
582-
NRepeat,
583-
KPack,
584-
KInner>())>;
562+
using BlockwiseGemmPipe =
563+
remove_cvref_t<decltype(BlockGemmPipeline_Selector<BlkGemmPipelineVer,
564+
BlkGemmPipeSched,
565+
BlockSize,
566+
LDSTypeA,
567+
LDSTypeB,
568+
ComputeTypeA,
569+
ComputeTypeB,
570+
AccDataType,
571+
decltype(MakeAWmmaTileDescriptor()),
572+
decltype(MakeBWmmaTileDescriptor()),
573+
ABlockTransferSrcScalarPerVector,
574+
BBlockTransferSrcScalarPerVector,
575+
MPerBlock,
576+
NPerBlock,
577+
KPerBlock,
578+
MPerWmma,
579+
NPerWmma,
580+
MRepeat,
581+
NRepeat,
582+
KPack,
583+
KInner>())>;
585584

586585
// Used to create obj in global function and pass it to Run method
587586
using EpilogueCShuffle =

0 commit comments

Comments
 (0)