Only fuse pointwise with layernorm on the first argument (#3679)

ROCm · Dec 12, 2024 · b72708a · b72708a
1 parent 7eef2b3
commit b72708a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/targets/gpu/fuse_ops.cpp b/src/targets/gpu/fuse_ops.cpp
@@ -835,7 +835,7 @@ struct find_layernorm_pointwise
 {
     auto matcher() const
     {
-        return precompile_name("pointwise")(match::any_of[match::inputs()](
+        return precompile_name("pointwise")(match::arg(0)(
             precompile_name("gpu::prelayernorm", "gpu::preadd_layernorm").bind("layernorm")));
     }
 

diff --git a/test/gpu/fuse_ops.cpp b/test/gpu/fuse_ops.cpp
@@ -103,8 +103,7 @@ TEST_CASE(layernorm_pointwise)
     {
         migraphx::program p1 = create_program(false);
         run_pass(p1);
-        migraphx::program p2 = create_fused_program();
-        EXPECT(p1 == p2);
+        EXPECT(p1 == create_program(false));
     }
 }