Ke Wen 51ee9b154e [c10d] Add NCCL memory allocator (#145675) ... This PR implements a small UI improvement over #133603. It prepares a NCCL memory allocator in torch cpp and then pybind's it out, so that user can directly use it. UI: ``` pool = torch.cuda.MemPool(backend.mem_allocator) with torch.cuda.use_mem_pool(pool): tensor = torch.arange(1024 * 1024 * 2, device=device) ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/145675 Approved by: https://github.com/syed-ahmed, https://github.com/wconstab		2025-01-30 18:19:00 +00:00
..
_composable	Revert "pickler for GraphModule (#141659)"	2025-01-27 22:39:30 +00:00
_shard	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
_tools	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
algorithms	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
bin
checkpoint	[dcp] Add ZStandard transformer (#143360)	2025-01-25 00:14:07 +00:00
elastic	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
flight_recorder	Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"	2025-01-04 14:17:20 +00:00
fsdp	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
launcher	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
nn/jit	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
optim	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
pipelining	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
rpc
tensor	[DTensor] Add pointwise ops strategy for aten.minimum (#145816)	2025-01-29 01:19:01 +00:00
argparse_util_test.py
test_backends.py
test_c10d_common.py	[BE][Ez]: FURB148 - remove useless enumerate calls (#145619)	2025-01-24 23:37:15 +00:00
test_c10d_functional_native.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_c10d_gloo.py	[ROCm] Enable post-merge trunk workflow on MI300 runners; skip and fix MI300 related failed tests (#143673)	2025-01-09 05:18:57 +00:00
test_c10d_logger.py
test_c10d_nccl.py	[c10d] Add NCCL memory allocator (#145675)	2025-01-30 18:19:00 +00:00
test_c10d_object_collectives.py
test_c10d_ops_nccl.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_c10d_pypg.py	DistributedDataParallel: add init_sync option to control collectives during initialization (#142824)	2024-12-11 20:28:38 +00:00
test_c10d_spawn.py
test_c10d_spawn_gloo.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_c10d_spawn_nccl.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_c10d_spawn_ucc.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_c10d_ucc.py	XFAIL test_save_load_checkpoint (#144927)	2025-01-16 07:31:56 +00:00
test_collective_utils.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_composability.py	composability test cleanup (#145011)	2025-01-18 04:37:12 +00:00
test_compute_comm_reordering.py
test_control_collectives.py
test_data_parallel.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_device_mesh.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_distributed_spawn.py
test_dynamo_distributed.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_fake_pg.py	[BE][Ez]: FURB148 - remove useless enumerate calls (#145619)	2025-01-24 23:37:15 +00:00
test_functional_api.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_inductor_collectives.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_launcher.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_multi_threaded_pg.py
test_nccl.py
test_pg_wrapper.py
test_store.py	[BE][CI] bump ruff to 0.8.4 (#143753)	2024-12-24 12:24:10 +00:00
test_symmetric_memory.py	[SymmetricMemory] fix an issue where rendezvous is performed with wrong device context when torch.cuda.set_device() is not callled (#144886)	2025-01-28 01:43:37 +00:00