acro bug fix

9170835c · Rich Ho · 17e2a5c5 · 9170835c
Commit 9170835c authored Jun 28, 2021 by Rich Ho
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

cuda/global_exchange.cpp cuda/global_exchange.cpp +1 -1

No files found.
--- a/cuda/global_exchange.cpp
+++ b/cuda/global_exchange.cpp
@@ -81,7 +81,7 @@ public:
        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,