pytorch/torch/csrc/cuda
Jane Xu 533cb9530e Introducing TORCH_CUDA_CPP_API and TORCH_CUDA_CU_API to the code (#50627)
Summary:
Sub-step of my attempt to split up the torch_cuda library, as it is huge. Please look at https://github.com/pytorch/pytorch/issues/49050 for details on the split and which files are in which target.

This PR introduces two new macros for Windows DLL purposes, TORCH_CUDA_CPP_API and TORCH_CUDA_CU_API. Both are defined as TORCH_CUDA_API for the time being.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/50627

Reviewed By: mruberry

Differential Revision: D25955441

Pulled By: janeyx99

fbshipit-source-id: ff226026833b8fb2fb7c77df6f2d6c824f006869
2021-01-21 19:09:11 -08:00
..
shared Make cudaHostRegister actually useful on cudart. (#45159) 2020-09-23 11:05:44 -07:00
comm.cpp Add torch::cuda::ncll::all2all (#45900) 2020-10-07 23:56:31 -07:00
comm.h Introducing TORCH_CUDA_CPP_API and TORCH_CUDA_CU_API to the code (#50627) 2021-01-21 19:09:11 -08:00
device_set.h
Event.cpp Remove PyCFunction casts as much as possible. (#46227) 2020-10-20 15:01:51 -07:00
Event.h
Graph.cpp [CUDA graphs] Cuda RNG-safe graph capture and replay bindings (#48875) 2020-12-14 10:51:58 -08:00
Module.cpp Add torch.cuda.can_device_access_peer (#50446) 2021-01-12 20:30:45 -08:00
Module.h
nccl.cpp Add support for NCCL alltoall (#44374) 2021-01-20 14:57:12 -08:00
nccl.h Introducing TORCH_CUDA_CPP_API and TORCH_CUDA_CU_API to the code (#50627) 2021-01-21 19:09:11 -08:00
override_macros.h
python_comm.cpp Add out= variants for cuda.comm.broadcast/gather/scatter (#39681) 2020-06-24 12:58:19 -07:00
python_comm.h
python_nccl.cpp Enable torch.cuda.nccl typechecking (#45344) 2020-09-25 17:02:47 -07:00
python_nccl.h
restore_macros.h
serialization.cpp
serialization.h
Storage.cpp Do not use PRId64 in torch/csrc (#44767) 2020-09-17 14:00:02 -07:00
Storage.h
Stream.cpp PyLong_{As/From}{Long/UnsignedLong} lint checks (#49280) 2020-12-17 09:32:08 -08:00
Stream.h Define the record_stream method in native_functions.yaml (#44301) 2020-10-13 09:15:22 -07:00
Tensor.cpp
THCP.h
undef_macros.h
utils.cpp
utils.h