Merge branch 'dot-add' into bert-opt2

3b8ae098 · Paul · ba4b69a7 · cf9cec1c · 3b8ae098
Commit 3b8ae098 authored Jul 01, 2022 by Paul
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

src/targets/gpu/fuse_ops.cpp src/targets/gpu/fuse_ops.cpp +3 -2

No files found.
--- a/src/targets/gpu/fuse_ops.cpp
+++ b/src/targets/gpu/fuse_ops.cpp
@@ -976,8 +976,9 @@ struct find_gemm_pointwise
        return precompile_name("pointwise")(
            match::nargs(3),
            match::all_of[match::inputs()](match::standard_shape()),
-            match::either_arg(0, 1)(match::used_once().bind("c"),
+            match::either_arg(0, 1)(
-                                    match::name("gpu::gemm")(match::nargs(3)).bind("gemm")));
+                match::any().bind("c"),
+                match::name("gpu::gemm")(match::nargs(3), match::used_once()).bind("gemm")));
    }
    // TODO: Move to matcher.hpp