Update device_batched_gemm_xdl.hpp

bda56e70 · zjing14 · GitHub · dae1c624 · bda56e70
Unverified Commit bda56e70 authored Sep 05, 2023 by zjing14 Committed by GitHub Sep 05, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

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,
+        ADataType, // TODO: distinguish A/B datatype
        AccDataType,
        CDataType,
        InMemoryDataOperationEnum::Set,