fixed mnk padding

dae1c624 · Jing Zhang · 5393111f · dae1c624
Commit dae1c624 authored Sep 06, 2023 by Jing Zhang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

include/ck/tensor_operation/gpu/device/impl/device_batched_gemm_xdl.hpp ...sor_operation/gpu/device/impl/device_batched_gemm_xdl.hpp +2 -2

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_batched_gemm_xdl.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_batched_gemm_xdl.hpp
@@ -175,7 +175,7 @@ struct DeviceBatchedGemmXdl : public DeviceBatchedGemm<ALayout,
    // GridwiseGemm
    using GridwiseGemm = GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r3_ext<
        BlockSize,
-        ADataType, // TODO: distinguish A/B datatype
+        ADataType,
        AccDataType,
        CDataType,
        InMemoryDataOperationEnum::Set,
@@ -185,7 +185,7 @@ struct DeviceBatchedGemmXdl : public DeviceBatchedGemm<ALayout,
        AElementwiseOperation,
        BElementwiseOperation,
        CElementwiseOperation,
-        GemmSpecialization::MNPadding,
+        GemmSpecialization::MNKPadding,
        MPerBlock,
        NPerBlock,
        K0PerBlock,