Remove used_once from find_add_layernorm

ddfac084 · Alan Turner · f4a1e94d · ddfac084
Commit ddfac084 authored Feb 07, 2023 by Alan Turner
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/targets/gpu/prefuse_ops.cpp src/targets/gpu/prefuse_ops.cpp +2 -1

No files found.
--- a/src/targets/gpu/prefuse_ops.cpp
+++ b/src/targets/gpu/prefuse_ops.cpp
@@ -100,8 +100,9 @@ struct find_add_layernorm
 {
    auto matcher() const
    {
+        // CK branch does not work with the used_once() limitation
        return match::layernorm()(
-            match::var("x")(match::name("add")(match::used_once()).bind("add")));
+            match::var("x")(match::name("add").bind("add")));
    }

    void apply(module& m, const match::matcher_result& r) const