pytorch
cc54da48 - Inductor cpp wrapper: fix FallbackKernel support (#100788)

Commit
1 year ago
Inductor cpp wrapper: fix FallbackKernel support (#100788) Fixes cpp wrapper support for kernels that are not exposed in `torch.ops.aten`. The current PR limits the support scope to `repeat_interleave.Tensor` and will submit follow-up PRs for more OPs. The PR maps the python schema of the kernel to the cpp schema and uses `c10::Dispatcher::singleton().findSchemaOrThrow` to find the corresponding cpp OP. The current support is limited and will raise `AssertionError` for unsupported cases. The limitation includes: - only support kernel that is not alias - only support kernel the args and returns of which don't have `alias_info` - only support output args to be a `Tensor` - only support input args to be `Tensor`, `Optional[int]`, `Optional[float]` and `Optional[bool]` Pull Request resolved: https://github.com/pytorch/pytorch/pull/100788 Approved by: https://github.com/jgong5, https://github.com/desertfire
Author
Committer
Parents
Loading