pytorch
b00a5359 - Add a way to skip lowering to nvprims (#85811)

Commit
3 years ago
Add a way to skip lowering to nvprims (#85811) This PR adds `skip_ops` argument to `TorchRefsNvfuserCapabilityMode` and `NvfuserPrimsMode` which is an iterable of function names to be skipped in the translation to nvprims process. Pull Request resolved: https://github.com/pytorch/pytorch/pull/85811 Approved by: https://github.com/mruberry, https://github.com/jjsjann123
Author
Committer
Parents
Loading