update

Chao1Han · Sep 5, 2024 · 4f73180 · 4f73180
1 parent fb9746b
commit 4f73180
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/torch/csrc/distributed/c10d/Ops.cpp b/torch/csrc/distributed/c10d/Ops.cpp
@@ -516,6 +516,7 @@ REGISTER_C10D_OP(alltoall_)
 REGISTER_C10D_OP(alltoall_base_)
 REGISTER_C10D_OP(barrier)
 
+REGISTER_C10D_OP1(allreduce_, XPU)
 // The following ops are specialized, register them separately
 
 TORCH_LIBRARY_IMPL(c10d, CPU, m) {

diff --git a/torch/csrc/distributed/c10d/ProcessGroupXCCL.hpp b/torch/csrc/distributed/c10d/ProcessGroupXCCL.hpp
@@ -38,7 +38,7 @@ using XCCL_KVS = ccl::shared_ptr_class<ccl::kvs>;
 constexpr const char* XCCL_BACKEND_NAME = "xccl";
 // using namespace torch::xpu::xccl;
 
-class TORCH_XPU_API ProcessGroupXCCL : public Backend {
+class TORCH_API ProcessGroupXCCL : public Backend {
  public:
   class WorkXCCL : public Work {
    public: