pytorch
f21a176c - Python Dispatcher should respect FuncTorchBatchedDecomposition key (#98328)

Commit
2 years ago
Python Dispatcher should respect FuncTorchBatchedDecomposition key (#98328) Fixes https://github.com/pytorch/pytorch/issues/97425. Python Dispatcher's resolve_key function should be equivalent to computeDispatchTableEntryWithDebug. We added a section to computeDispatchTableEntryWithDebug but forgot to add it to resolve_key. This PR fixes that discrepancy. Test Plan: - new test Pull Request resolved: https://github.com/pytorch/pytorch/pull/98328 Approved by: https://github.com/Chillee, https://github.com/kshitij12345, https://github.com/Neilblaze
Author
Committer
Parents
Loading