Add missing ComputeDatatype in contraction_multi_ABD_xdl_fp16 (#1024)

16eb824c · Bartlomiej Wroblewski · GitHub · 4ef704d8 · 16eb824c
Unverified Commit 16eb824c authored Nov 03, 2023 by Bartlomiej Wroblewski Committed by GitHub Nov 03, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp ..._contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp +2 -0

No files found.
--- 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>;