pytorch
74877943 - Don't invoke mode as overloaded argument in torch dispatch (#80992)

Commit
2 years ago
Don't invoke mode as overloaded argument in torch dispatch (#80992) I noticed that in some situations torch dispatch modes were being invoked with a mode active, which isn't supposed to happen (we disable modes before calling into the user mode.) I also noticed that I was getting a warning that I had a deprecated non-static definition of torch dispatch on an argument even though there wasn't any. It turns out this is because modes were part of the overloaded arguments list in the Python fallback kernel for torch dispatch. This is wrong; instead we should rely on the actual dispatching function to consult modes. This makes the code simpler. Signed-off-by: Edward Z. Yang <ezyang@fb.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/80992 Approved by: https://github.com/zou3519
Author
Committer
Parents
Loading