pytorch/torch/csrc/cuda
2025-10-19 15:34:36 +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 [Mem Snapshot] Add Metadata Field (#165490) 2025-10-17 23:46:02 +00:00
memory_snapshot.h [Memory Snapshot] Add Flag to Toggle Global and Local Callbacks for Annotations (#154932) 2025-06-04 23:15:19 +00:00
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 Introduce a generic API torch._C._accelerator_setAllocatorSettings (#165291) 2025-10-19 15:34:36 +00:00
Module.h
nccl.cpp [Code Clean] Replace std::runtime_error with TORCH_CHECK (#163437) 2025-10-12 01:23:02 +00:00
nccl.h
python_comm.cpp
python_comm.h
python_nccl.cpp Turn some const strings into constexpr in C++ code (#165203) 2025-10-13 20:25:20 +00:00
python_nccl.h
Stream.cpp
Stream.h
THCP.h Make torch/csrc/utils.h to be device-agnostic (#152521) 2025-05-04 07:15:11 +00:00
utils.cpp [Code Clean] Replace std::runtime_error with TORCH_CHECK (#163437) 2025-10-12 01:23:02 +00:00
utils.h Make torch/csrc/utils.h to be device-agnostic (#152521) 2025-05-04 07:15:11 +00:00