Mark CUDA 10.1 as unsupported. (#4265)

This commit is contained in:
Jiaming Yuan
2019-03-17 16:59:15 +08:00
committed by GitHub
parent fdcae024e7
commit cf8d5b9b76
2 changed files with 9 additions and 7 deletions

View File

@@ -26,8 +26,9 @@
#include "../common/io.h"
#endif
// Uncomment to enable
#define TIMERS
#if __CUDACC_VER_MAJOR__ == 10 && __CUDACC_VER_MINOR__ == 1
#error "CUDA 10.1 is not supported, see #4264."
#endif
namespace dh {
@@ -893,14 +894,14 @@ class AllReducer {
int nccl_nranks = std::accumulate(device_counts.begin(),
device_counts.end(), 0);
nccl_rank += nccl_rank_offset;
GroupStart();
for (size_t i = 0; i < device_ordinals.size(); i++) {
int dev = device_ordinals.at(i);
dh::safe_cuda(cudaSetDevice(dev));
dh::safe_nccl(ncclCommInitRank(
&comms.at(i),
nccl_nranks, id,
nccl_nranks, id,
nccl_rank));
nccl_rank++;