pytorch/torch/_prims
Ivan Yashchuk cd7c86eaa4 Add prims.clone (#86705)
This simple PR adds `clone` as a primitive.
Current implementation of `clone` is not supported with nvFuser executor because of `empty_like` + `copy_to`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/86705
Approved by: https://github.com/mruberry
2022-10-12 18:22:00 +00:00
..
__init__.py Add prims.clone (#86705) 2022-10-12 18:22:00 +00:00
context.py Add Skip List for Aten Ops that are fused in nvFuser. (#86101) 2022-10-07 03:55:13 +00:00
executor.py Add config to PrimTorch's nvFuser executor (#84482) 2022-09-09 07:58:21 +00:00
nvfuser_executor.py Add nvFuser support for torch.native_batch_norm (#85562) 2022-10-03 15:03:08 +00:00
nvfuser_prims.py Add prims.clone (#86705) 2022-10-12 18:22:00 +00:00