We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 239a9ad commit 293fccfCopy full SHA for 293fccf
torch/csrc/cuda/nccl.h
@@ -75,7 +75,7 @@ enum class ncclDataType {
75
// RAII helper class to manage NCCL group API and CUDA free mutex.
76
// The destructor is allowed to throw since this helper class only
77
// manages group and lock lifetimes.
78
-struct AutoNcclGroup {
+struct TORCH_CUDA_CPP_API AutoNcclGroup {
79
AutoNcclGroup();
80
AutoNcclGroup(ncclComm_t comm, bool comm_nonblocking);
81
~AutoNcclGroup() noexcept(false);
0 commit comments