Fix bug in unsqueeze_nested stride calculation (#88688)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/88688
Approved by: https://github.com/cpuhrsch
This commit is contained in:
Mikayla Gawarecki
2023-02-10 08:02:32 +00:00
committed by PyTorch MergeBot
parent 889a4640a0
commit c7c7238976
4 changed files with 19 additions and 4 deletions

View File

@ -154,7 +154,6 @@ _SKIP_PYTHON_BINDINGS = [
"fill.Scalar", # only used by the functionalization pass
"lift.*",
"normal_functional", # only used by the functionalization pas
"_nested_tensor_strides", # don't want to expose this to python
"_nested_tensor_offsets", # don't want to expose this to python
"_nested_view_from_buffer", # View only version of _nested_from_buffer. This will force users to only use the "safe" version.
"_nested_view_from_buffer_copy",