Add _unsafe_view to tests

csarofeen · Sep 22, 2022 · 254161d · 254161d
1 parent 487a7a8
commit 254161d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_prims.py b/test/test_prims.py
@@ -538,10 +538,10 @@ def func4(a):
             return torch.reshape(a, tuple(reversed(a.shape)))
 
         def func5(a):
-            return torch.ops.aten.view(a, tuple(reversed(a.shape)))
+            return torch.ops.aten.view.default(a, tuple(reversed(a.shape)))
 
         def func6(a):
-            return torch.ops.aten.view.default(a, tuple(reversed(a.shape)))
+            return torch.ops.aten._unsafe_view.default(a, tuple(reversed(a.shape)))
 
         for func in (func1, func2, func3, func4, func5, func6):
             with TorchRefsNvfuserCapabilityMode():