Add _unsafe_view to list

csarofeen · Sep 22, 2022 · 79c18da · 79c18da
1 parent 254161d
commit 79c18da
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/_prims/context.py b/torch/_prims/context.py
@@ -238,6 +238,7 @@ def _is_view_or_reshape(self, func):
             or "torch.view_copy" == torch.overrides.resolve_name(func)
             or "torch.reshape" == torch.overrides.resolve_name(func)
             or func == torch.ops.aten.view.default
+            or func == torch.ops.aten._unsafe_view.default
         )
 
     def __torch_function__(