From 47dcf0940febb7b8045ea48a098791f1f97acb0b Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Wed, 20 Aug 2025 13:34:27 +0000
Subject: [PATCH] Complete fix for duplicate torch operations - all ops use
 unique library names

Co-authored-by: ProExpertProg <11367180+ProExpertProg@users.noreply.github.com>
---
 tests/compile/piecewise/test_multiple_graphs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/compile/piecewise/test_multiple_graphs.py b/tests/compile/piecewise/test_multiple_graphs.py
index b71ae66d3fd28..a5e18a2c0dcec 100644
--- a/tests/compile/piecewise/test_multiple_graphs.py
+++ b/tests/compile/piecewise/test_multiple_graphs.py
@@ -90,7 +90,7 @@ class Attention(nn.Module):
         x = self.pre_attn(x)
         x = self.rms_norm_ref(x)
         attn_output = torch.empty_like(x)
-        torch.ops.silly.attention(x, x, x, attn_output)
+        torch.ops.silly_multiple.attention(x, x, x, attn_output)
         x = attn_output
         x = self.rms_norm_ref(x)
         x = self.post_attn(x)