pytorch
d2e18606 - Fix view issue in embedding_dense_backward decomp (#79857)

Commit
2 years ago
Fix view issue in embedding_dense_backward decomp (#79857) I was hitting: ``` File "/home/jansel/pytorch/torch/fx/experimental/proxy_tensor.py", line 66, in proxy_call return CURRENT_DECOMPOSITION_TABLE[func_overload](*args, **kwargs) File "/home/jansel/pytorch/torch/_decomp/decompositions.py", line 801, in embedding_dense_backward indices_rank1 = indices.view(numel) File "/home/jansel/pytorch/torch/fx/experimental/proxy_tensor.py", line 122, in __torch_dispatch__ return proxy_call(func_overload, args, kwargs) File "/home/jansel/pytorch/torch/fx/experimental/proxy_tensor.py", line 86, in proxy_call real_out = func_overload(*args, **kwargs) File "/home/jansel/pytorch/torch/_ops.py", line 49, in __call__ return self._op(*args, **kwargs or {}) RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead. ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/79857 Approved by: https://github.com/Chillee
Author
Committer
Parents
Loading