pytorch
4eec865f - [nvFuser] Improving bitwise ops support (#77158)

Commit
2 years ago
[nvFuser] Improving bitwise ops support (#77158) - Some renaming to better match PyTorch API: - `lshift` -> `bitwise_left_shift` - `rshift` -> `bitwise_right_shift` - `andOp` -> `bitwise_and` - `orOp` -> `bitwise_or` - `xorOp` -> `bitwise_xor` - `notOp` -> `bitwise_not` - Fix type inferences and type checking of these ops - Add `bitwise_*` to parser and python frontend - Improve test coverage Pull Request resolved: https://github.com/pytorch/pytorch/pull/77158 Approved by: https://github.com/kevinstephano, https://github.com/jjsjann123
Author
Committer
Parents
Loading