Only gemm used once not c matrix

48cf58f6 · Paul · d6523b49 · 48cf58f6
Commit 48cf58f6 authored Jul 01, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/src/targets/gpu/fuse_ops.cpp
+++ b/src/targets/gpu/fuse_ops.cpp
@@ -975,8 +975,8 @@ 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::any().bind("c"),
-                                    match::name("gpu::gemm")(match::nargs(3)).bind("gemm")));
+                                    match::name("gpu::gemm")(match::nargs(3), match::used_once()).bind("gemm")));
    }
    // TODO: Move to matcher.hpp