Skip to content

GitLab

"tests/git@developer.sourcefind.cn:norm/vllm.git" did not exist on "7d2dcce175cec00bc1d127d6b3a5f1ef73a6ba3c"
Unverified Commit bda56e70 authored by zjing14's avatar zjing14 Committed by GitHub
Browse files

Update device_batched_gemm_xdl.hpp

parent dae1c624
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
View file @ bda56e70
...@@ -175,7 +175,7 @@ struct DeviceBatchedGemmXdl : public DeviceBatchedGemm<ALayout, ...@@ -175,7 +175,7 @@ struct DeviceBatchedGemmXdl : public DeviceBatchedGemm<ALayout,
// GridwiseGemm // GridwiseGemm
using GridwiseGemm = GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r3_ext< using GridwiseGemm = GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r3_ext<
BlockSize, BlockSize,
ADataType, ADataType, // TODO: distinguish A/B datatype
AccDataType, AccDataType,
CDataType, CDataType,
InMemoryDataOperationEnum::Set, InMemoryDataOperationEnum::Set,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment