pytorch/torch/csrc/cuda
2025-10-07 20:51:22 +00:00
..
shared Revert "Add magic TORCH_MAKE_PYBIND_ENUM_FASTER macro (#163527)" 2025-10-02 15:42:42 +00:00
comm.cpp
comm.h
CUDAPluggableAllocator.cpp [PyTorch CCA] Add an API to get expandable segment sizes (#163771) 2025-10-01 02:16:58 +00:00
CUDAPluggableAllocator.h [PyTorch CCA] Add an API to get expandable segment sizes (#163771) 2025-10-01 02:16:58 +00:00
device_set.h
Event.cpp [2/N] Fix cppcoreguidelines-init-variables suppression (#146237) 2025-06-19 23:26:42 +00:00
Event.h
GdsFile.cpp [1/N] Fix clang-tidy readability checks (#164561) 2025-10-04 09:40:38 +00:00
GdsFile.h
Graph.cpp [CUDAGraph] Add getter for cuda graph exec (#161294) 2025-08-25 20:57:37 +00:00
memory_snapshot.cpp [CUDA] Reuse blocks with record_stream during CUDA Graph capture in the CUDACachingAllocator (#158352) 2025-09-04 17:21:26 +00:00
memory_snapshot.h
MemPool.cpp [nccl symm mem] don't use arg for mempool, correctly use symmetric registration in hooks (#161238) 2025-08-25 03:09:32 +00:00
Module.cpp Revert "[CUDA] Add experimental green context support for SM carveout (#159104)" 2025-10-07 20:51:22 +00:00
Module.h
nccl.cpp [c10d][BE][ez] Update tensor ptr inside nccl.cpp (#164276) 2025-09-30 22:05:12 +00:00
nccl.h
python_comm.cpp
python_comm.h
python_nccl.cpp
python_nccl.h
Stream.cpp
Stream.h
THCP.h
utils.cpp
utils.h