..
comm.cpp
New implementations of DeviceGuard, StreamGuard and MultiStreamGuard (with CUDA specializations) ( #13342 )
2018-11-11 12:11:10 -08:00
comm.h
Rewrite THPUtils_PySequence_to_CUDAStreamList to return vector<optional<CUDAStream>> ( #13125 )
2018-10-29 08:27:23 -07:00
cuda_check.h
Remove NO_* and WITH_* across codebase, except in setup.py ( #8555 )
2018-06-15 12:29:48 -04:00
device_set.h
Replace std::size_t with size_t ( #8093 )
2018-06-04 11:10:44 -04:00
Module.cpp
Delete a bunch of uses of getType in favor of TensorOptions.
2018-08-30 20:11:24 -07:00
Module.h
Add torch.cuda.get_device_name function ( #2540 )
2017-08-26 15:06:37 -04:00
nccl.cpp
New implementations of DeviceGuard, StreamGuard and MultiStreamGuard (with CUDA specializations) ( #13342 )
2018-11-11 12:11:10 -08:00
nccl.h
DDP multi-GPU segfault fix ( #13291 )
2018-10-31 00:43:42 -07:00
override_macros.h
Don't override Tensor, Storage macros defined outside torch/csrc in t… ( #8243 )
2018-06-07 16:10:10 -04:00
python_comm.cpp
Rewrite THPUtils_PySequence_to_CUDAStreamList to return vector<optional<CUDAStream>> ( #13125 )
2018-10-29 08:27:23 -07:00
python_comm.h
Move broadcast and broadcast_coalesced to C++
2018-01-18 11:16:45 +01:00
python_nccl.cpp
New implementations of DeviceGuard, StreamGuard and MultiStreamGuard (with CUDA specializations) ( #13342 )
2018-11-11 12:11:10 -08:00
python_nccl.h
Add workaround to fix include warnings in Python 2 builds. ( #6716 )
2018-04-24 12:30:19 -07:00
restore_macros.h
Don't override Tensor, Storage macros defined outside torch/csrc in t… ( #8243 )
2018-06-07 16:10:10 -04:00
serialization.cpp
Add workaround to fix include warnings in Python 2 builds. ( #6716 )
2018-04-24 12:30:19 -07:00
serialization.h
Add torch.save and torch.load
2016-08-23 07:51:55 -07:00
Storage.cpp
Don't call into Python during Storage destruction. ( #10407 )
2018-08-13 11:20:07 -07:00
Storage.h
Refactor _C extension to export some utilities
2016-09-21 08:36:54 -07:00
Stream.cpp
Move CUDAStreamInternals inside detail namespace. ( #14109 )
2018-11-19 17:05:46 -08:00
Stream.h
Make all of TH and THC C++. ( #6913 )
2018-04-28 07:45:02 -04:00
Tensor.cpp
Add workaround to fix include warnings in Python 2 builds. ( #6716 )
2018-04-24 12:30:19 -07:00
THCP.h
use at::Half in THC ( #11322 )
2018-09-12 17:39:37 -07:00
undef_macros.h
Don't override Tensor, Storage macros defined outside torch/csrc in t… ( #8243 )
2018-06-07 16:10:10 -04:00
utils.cpp
Rewrite THPUtils_PySequence_to_CUDAStreamList to return vector<optional<CUDAStream>> ( #13125 )
2018-10-29 08:27:23 -07:00
utils.h
Sparse Library ( #333 )
2017-01-05 00:43:41 +01:00