Update cpu_adam.py (#1968)

microsoft · May 19, 2022 · 737fee6 · 737fee6
1 parent 7924942
commit 737fee6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/op_builder/cpu_adam.py b/op_builder/cpu_adam.py
@@ -20,7 +20,8 @@ def sources(self):
 
     def libraries_args(self):
         args = super().libraries_args()
-        args += ['curand']
+        if not self.is_rocm_pytorch():
+            args += ['curand']
         return args
 
     def include_paths(self):