Merge branch 'fix_contraction_abd_example' into 3d_grouped_conv_fp16_comp_fp8

dd410bbf · Jing Zhang · 80abbcff · 1b1a0197 · dd410bbf
Commit dd410bbf authored Oct 03, 2023 by Jing Zhang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp ..._contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp +1 -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
@@ -291,6 +291,7 @@ int main(int argc, char* argv[])
                                                                      BDataType,
                                                                      CShuffleDataType,
                                                                      AccDataType,
+                                                                      A0DataType,
                                                                      PassThrough,
                                                                      BElementOp>;