pytorch/torch/_prims
Ryan Spring 1d0d506e97 Add Div reference (#77936)
Add Prims:
-  trunc
-  Replace _wrap_scalar with scalar_tensor

Add Reference:
-  copysign
- div
- floor_divide
- trunc_divide

Other:
* Add support for `variant_test_name` in _find_referenced_opinfo
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77936
Approved by: https://github.com/mruberry
2022-06-27 14:46:17 +00:00
..
__init__.py Add Div reference (#77936) 2022-06-27 14:46:17 +00:00
context.py Fix interpretation torch -> torch._refs in case of nested torch calls under TorchRefsMode (#80135) 2022-06-25 03:55:04 +00:00
executor.py small cleanup of executor (#79973) 2022-06-22 00:35:51 +00:00
utils.py Fixes maybe_broadcast to actually broadcast only when needed (#79298) 2022-06-21 19:26:40 +00:00
wrappers.py linalg_cholesky_ex meta function 2022-06-03 23:11:02 +00:00