From a85ef2a95e1fc40b500d8cc5ba0f38c387f35c91 Mon Sep 17 00:00:00 2001 From: Bartlomiej Wroblewski Date: Fri, 3 Nov 2023 16:22:11 +0100 Subject: [PATCH] Add missing ComputeDatatype in contraction_multi_ABD_xdl_fp16 (#1024) [ROCm/composable_kernel commit: 16eb824c90b397722024e494daee75dd9c7e757c] --- .../61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp b/example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp index 6634dacd25..4317484336 100644 --- a/example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp +++ b/example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp @@ -34,6 +34,7 @@ using AccDataType = F32; using CShuffleDataType = F32; using DDataType = F16; using EDataType = F16; +using ComputeDataType = F16; static constexpr ck::index_t NumDimM = 2; static constexpr ck::index_t NumDimN = 2; @@ -291,6 +292,7 @@ int main(int argc, char* argv[]) BDataType, CShuffleDataType, AccDataType, + ComputeDataType, PassThrough, BElementOp>;