pytorch
f3d7ab54 - Unconditionally register Python decomps to Meta key in Python Dispatcher (#85750)

Commit
3 years ago
Unconditionally register Python decomps to Meta key in Python Dispatcher (#85750) This makes them available for Python Dispatcher to service them when symbolic shapes are involved. This is needed because under certain conditions, functionalization will directly call the Meta kernel for a function in order to produce a properly sized output wrapper tensor for a view operation. This direct call bypasses the normal decomposition table mechanism. Signed-off-by: Edward Z. Yang <ezyang@fb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/85750 Approved by: https://github.com/wconstab
Author
Committer
Parents
Loading