pytorch/torch/_prims
2022-11-20 00:42:03 +00:00
..
__init__.py Symintify numel(), infer_size, prims.elementwise_meta (#88956) 2022-11-20 00:42:03 +00:00
context.py [primTorch] Add ref for complex (#88562) 2022-11-13 20:31:16 +00:00
executor.py Add config to PrimTorch's nvFuser executor (#84482) 2022-09-09 07:58:21 +00:00
nvfuser_executor.py Add NVTX markers that dump additional information for nvprim_nvfuser Dynamo graphs (#88259) 2022-11-18 22:36:08 +00:00
nvfuser_prims.py Add min cut partitioner for AOT+nvFuser (#88204) 2022-11-09 12:56:55 +00:00