Update to fix rccl error in Microsoft CI

TedThemistokleous · Jul 10, 2024 · 4e9996e · 4e9996e
1 parent ddb339b
commit 4e9996e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmake/CMakeLists.txt b/cmake/CMakeLists.txt
@@ -1554,7 +1554,7 @@ if (UNIX AND onnxruntime_USE_NCCL)
   if (onnxruntime_USE_NCCL)
     if (onnxruntime_USE_CUDA)
       set(NCCL_LIBNAME "nccl")
-    elseif (onnxruntime_USE_ROCM)
+    elseif (onnxruntime_USE_ROCM) OR (onnxruntime_USE_MIGRAPHX)
       set(NCCL_LIBNAME "rccl")
     endif()
     find_path(NCCL_INCLUDE_DIR