From 79c18da42aba57eb6aa743dffc8e428c96b6223a Mon Sep 17 00:00:00 2001 From: Ivan Yashchuk Date: Thu, 22 Sep 2022 18:08:25 +0300 Subject: [PATCH] Add _unsafe_view to list --- torch/_prims/context.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torch/_prims/context.py b/torch/_prims/context.py index fa5a97df460ee..125ad065c9518 100644 --- 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__(