commit d8ad0ba5c1a90f24a651d8b674b16febdd1bba5b
author Rohan Varma <rvarm1@fb.com> Wed Oct 11 12:36:48 2023 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Thu Oct 12 08:54:32 2023 +0000
tree 147e1bf400de431609b1d7a51dfa6972ab417f96
parent b35279dfac2950c053d005fdbe179312c5f9d921

[Dist][ez][nit] Formatted nccl version string in startup (#111076)

Formats the string using the existing getNCCLversion

Differential Revision: [D50193558](https://our.internmc.facebook.com/intern/diff/D50193558/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/111076
Approved by: https://github.com/Skylion007

torch/csrc/distributed/c10d/ProcessGroupNCCL.cpp

diff --git a/torch/csrc/distributed/c10d/ProcessGroupNCCL.cpp b/torch/csrc/distributed/c10d/ProcessGroupNCCL.cpp
index ca8dd08..6dfdd0a 100644
--- a/torch/csrc/distributed/c10d/ProcessGroupNCCL.cpp
+++ b/torch/csrc/distributed/c10d/ProcessGroupNCCL.cpp
@@ -690,13 +690,8 @@
       parseEnvVarString("TORCH_DISTRIBUTED_DEBUG", OFF.c_str());
   const char* nccl_debug = parseEnvVarString("NCCL_DEBUG", OFF.c_str());
   LOG(INFO) << "[Rank " << rank_
-            << "] ProcessGroupNCCL initialization options:"
-#ifdef NCCL_VERSION_CODE
-            // NCCL > 2.3.5 should have NCCL_VERSION_CODE defined
-            << "NCCL VERSION: " << NCCL_VERSION_CODE
-#else
-            << "NCCL VERSION: " << NCCL_MAJOR << NCCL_MINOR << NCCL_PATCH
-#endif
+            << "] ProcessGroupNCCL initialization options: "
+            << "NCCL version: " << getNcclVersion()
             << ", NCCL_ASYNC_ERROR_HANDLING: " << asyncErrorHandling_
             << ", NCCL_DESYNC_DEBUG: " << desyncDebug_
             << ", NCCL_ENABLE_TIMING: " << enableTiming_.load()