Update on "Enable TensorPipe CUDA sending to self"

Differential Revision: [D25941964](https://our.internmc.facebook.com/intern/diff/D25941964) [ghstack-poisoned]
pytorch · Jan 18, 2021 · ed9ede9 · ed9ede9
1 parent a181e7a
commit ed9ede9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch/csrc/distributed/rpc/tensorpipe_agent.cpp b/torch/csrc/distributed/rpc/tensorpipe_agent.cpp
@@ -85,6 +85,9 @@ constexpr int64_t kBasicChannelPriority = 0;
 
 #if TENSORPIPE_HAS_CUDA_IPC_CHANNEL && defined(USE_CUDA_NOT_ROCM)
 constexpr int64_t kCudaIpcChannelPriority = 300;
+#endif
+
+#ifdef USE_CUDA_NOT_ROCM
 constexpr int64_t kCudaXthChannelPriority = 400;
 #endif