acro bug fix

laekov · Jun 28, 2021 · 9170835 · 9170835
1 parent 17e2a5c
commit 9170835
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cuda/global_exchange.cpp b/cuda/global_exchange.cpp
@@ -81,7 +81,7 @@ class HackNCCLGroup: public c10d::ProcessGroupNCCL {
         if (rank == 0) {
             ncclGetUniqueId(&ncclID);
         }
-#if defined(TORCH_VERSION) && (TORCH_VERSION_MAJOR > 1 || \
+#if defined(TORCH_VERSION_MAJOR) && (TORCH_VERSION_MAJOR > 1 || \
         (TORCH_VERSION_MAJOR == 1 && TORCH_VERSION_MINOR >= 8))
         broadcastUniqueNCCLID(&ncclID,
                 c10d::OpType::SEND,