Complete fix for duplicate torch operations - all ops use unique library names

Co-authored-by: ProExpertProg <11367180+ProExpertProg@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot] 2025-08-20 13:34:27 +00:00
parent 9d6f0372e5
commit 47dcf0940f

View File

@ -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)