update

Chao1Han · Oct 15, 2024 · 5968f0f · 5968f0f
1 parent b57e812
commit 5968f0f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cmake/Modules/FindXCCL.cmake b/cmake/Modules/FindXCCL.cmake
@@ -8,8 +8,8 @@ include(FindPackageHandleStandardArgs)
 
 set(XCCL_ROOT "/opt/intel/oneapi/ccl/latest")
 if (NOT EXISTS "${XCCL_ROOT}")
-  message(STATUS "Default OneCCL not found, using current environment OneCCL")
-  set(XCCL_ROOT $ENV{CCL_ROOT})
+  message(STATUS "Default OneCCL not found, using current environment OneAPI")
+  set(XCCL_ROOT $ENV{ONEAPI_ROOT}/ccl/latest)
 endif()
 
 string(COMPARE EQUAL "${XCCL_ROOT}" "" nocclfound)

diff --git a/torch/csrc/distributed/c10d/ProcessGroupXCCL.cpp b/torch/csrc/distributed/c10d/ProcessGroupXCCL.cpp
@@ -184,7 +184,8 @@ std::shared_ptr<xcclComm_t> ProcessGroupXCCL::getXCCLComm(
   rank = getRank();
 
   c10::impl::VirtualGuardImpl impl(device.type());
-  c10::Stream stream = impl.getStream(device);
+  c10::Stream stream =
+      impl.getStreamFromGlobalPool(device, /*isHighPriority=*/false);
   sycl::queue& q = c10::xpu::XPUStream(stream).queue();
 
   auto ctx = ccl::create_context(q.get_context());