OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Skip storage check debug assert in view codegen when output is a subclass instance (#122718)

Browse Source 
Before the fix, this assert blows up in DEBUG mode for views where the input (base) is a dense tensor and the output (view) is a subclass instance.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/122718
Approved by: https://github.com/soulitzer
This commit is contained in:
Joel Schlosser 2024-03-26 15:07:30 -04:00 committed by PyTorch MergeBot
parent 105381ea11
commit eb5381da66
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/autograd/gen_variable_type.py
View File

@ -417,7 +417,8 @@ ENFORCE_SAME_TENSOR_STORAGE = CodeTemplate(
"""\ """\
if (${tensor_name}_storage_saved.has_value() && if (${tensor_name}_storage_saved.has_value() &&
!at::impl::dispatch_mode_enabled() && !at::impl::dispatch_mode_enabled() &&
!at::impl::tensor_has_dispatch(${tensor_name})) !at::impl::tensor_has_dispatch(${tensor_name}) &&
!at::impl::tensor_has_dispatch(${out_tensor_name}))
TORCH_INTERNAL_ASSERT(${tensor_name}_storage_saved.value().is_alias_of(${out_tensor_name}.storage())); TORCH_INTERNAL_ASSERT(${tensor_name}_storage_saved.value().is_alias_of(${out_tensor_name}.storage()));
""" """
) )