Change source of CPUAdam for xpu accelerator (#5703)

Noted that cpu adam for cuda/cpu accelerator has removed the dependency of CUDA, we can now use the same source.
deepspeedai · Jun 28, 2024 · 4b8a4a0 · 4b8a4a0
1 parent f0e3f01
commit 4b8a4a0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 270 deletions.
diff --git a/csrc/xpu/adam/cpu_adam.cpp b/csrc/xpu/adam/cpu_adam.cpp
diff --git a/csrc/xpu/adam/cpu_adam_impl.cpp b/csrc/xpu/adam/cpu_adam_impl.cpp
diff --git a/op_builder/xpu/cpu_adam.py b/op_builder/xpu/cpu_adam.py
@@ -17,13 +17,11 @@ def absolute_name(self):
         return f'deepspeed.ops.adam.{self.NAME}_op'
 
     def sources(self):
-        if self.build_for_cpu:
-            return ['csrc/xpu/adam/cpu_adam.cpp', 'csrc/xpu/adam/cpu_adam_impl.cpp']
+        return ['csrc/adam/cpu_adam.cpp', 'csrc/adam/cpu_adam_impl.cpp']
 
-        return [
-            'csrc/xpu/adam/cpu_adam.cpp', 'csrc/xpu/adam/cpu_adam_impl.cpp',
-            'csrc/xpu/common/custom_cuda_kernel.dp.cpp'
-        ]
+    def libraries_args(self):
+        args = super().libraries_args()
+        return args
 
     def include_paths(self):
-        return ['csrc/xpu/includes']
+        return ['csrc/includes']