relax

Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
Lightning-AI · Jun 13, 2024 · fe589dc · fe589dc
1 parent a1704f3
commit fe589dc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/thunder/tests/test_core.py b/thunder/tests/test_core.py
@@ -2121,7 +2121,8 @@ def test_xor(s, o):
 
     for t in tests:
         cfn = thunder.jit(t)
-        with pytest.raises(RuntimeError, match="not supported"):
+        # Some ops of `tests` already have in-place supported, leading to broadcast error
+        with pytest.raises(RuntimeError, match="not supported|Attempting"):
             cfn(t1, t2)
         # Note: Python maps inplace operations on (immutuables) to
         #       out of place operations, NumberProxy does this, too.