pytorch/torch/csrc/cuda
2024-10-15 17:19:16 +00:00
..
shared [sparse] Add cuSPARSELt as a backend (#128534) 2024-08-21 22:06:07 +00:00
comm.cpp [2/N] Fix extra warnings brought by clang-tidy-17 (#137459) 2024-10-08 19:05:02 +00:00
comm.h [1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301) 2024-07-08 07:03:53 +00:00
CUDAPluggableAllocator.cpp [2/N] Fix extra warnings brought by clang-tidy-17 (#137459) 2024-10-08 19:05:02 +00:00
CUDAPluggableAllocator.h Revert "[ROCm] remove caffe2 from hipify (#137157)" 2024-10-08 17:45:45 +00:00
device_set.h
Event.cpp [BE][clang-format] make macro PyObject_HEAD_INIT(type) and PyVarObject_HEAD_INIT(type, size) have its own line (#136949) 2024-10-02 18:39:22 +00:00
Event.h Make device-specific event inherits from torch.Event (#134845) 2024-10-01 06:28:41 +00:00
GdsFile.cpp [Reland] Add wrappers for synchronous GPUDirect Storage APIs (#133489) 2024-08-15 17:11:52 +00:00
GdsFile.h [Reland] Add wrappers for synchronous GPUDirect Storage APIs (#133489) 2024-08-15 17:11:52 +00:00
Graph.cpp
memory_snapshot.cpp Revert "Make Context to be Device-agnostic Step by Step (1/N) (#136519)" 2024-10-15 17:19:16 +00:00
memory_snapshot.h [1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301) 2024-07-08 07:03:53 +00:00
MemPool.cpp Implements torch.cuda.MemPool() API (#131152) 2024-08-01 01:29:30 +00:00
Module.cpp Revert "Make Context to be Device-agnostic Step by Step (1/N) (#136519)" 2024-10-15 17:19:16 +00:00
Module.h
nccl.cpp [AMD] Do not skip 0-byte send/recv (#137952) 2024-10-15 09:35:03 +00:00
nccl.h add TORCH_CUDA_CPP_API for AutoNcclGroup (#130012) 2024-09-20 21:20:25 +00:00
python_comm.cpp
python_comm.h
python_nccl.cpp [1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301) 2024-07-08 07:03:53 +00:00
python_nccl.h
Stream.cpp [BE][clang-format] make macro PyObject_HEAD_INIT(type) and PyVarObject_HEAD_INIT(type, size) have its own line (#136949) 2024-10-02 18:39:22 +00:00
Stream.h
Tensor.cpp
THCP.h
utils.cpp