Skip to content

Commit

Permalink
Skip test_multiple_views_before_inplace_to_base for nvFuser.
Browse files Browse the repository at this point in the history
  • Loading branch information
wujingyue committed Aug 22, 2024
1 parent a96ebbb commit b18de98
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions thunder/tests/test_inplace_functionalization.py
Original file line number Diff line number Diff line change
Expand Up @@ -364,6 +364,14 @@ def f_with_view(a):
dtypes=NOTHING,
)
def test_multiple_views_before_inplace_to_base(executor, device, _):
from thunder.tests.framework import nvFuserTestExecutor

if type(executor) is nvFuserTestExecutor:
pytest.skip(
"nvFuser doesn't enforce the order between `z=x.view(-1)` and "
"`x.add_(1)`, so the behavior is undefined due to this "
"race condition."
)

# ref: https://github.com/pytorch/pytorch/blob/29e2e2a/test/test_functionalization.py#L159-L169
def f(x):
Expand Down

0 comments on commit b18de98

Please sign in to comment.