add default ccl root dir

Chao1Han · Oct 12, 2024 · f69059a · f69059a
1 parent 3d55b85
commit f69059a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmake/Modules/FindXCCL.cmake b/cmake/Modules/FindXCCL.cmake
@@ -6,8 +6,14 @@
 
 include(FindPackageHandleStandardArgs)
 
+# include FindMKL.cmake to use its variables
+include(FindMKL)
+
 set(XCCL_ROOT "")
-if(DEFINED ENV{CCL_ROOT})
+if (EXISTS "${DEFAULT_INTEL_COMPILER_DIR}/ccl/latest")
+  SET(XCCL_ROOT "${DEFAULT_INTEL_COMPILER_DIR}/ccl/latest")
+elseif(DEFINED ENV{CCL_ROOT})
+  message(STATUS "Default OneCCL not found, using current environment OneCCL")
   set(XCCL_ROOT $ENV{CCL_ROOT})
 endif()