| .. |
|
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
|
|
|