add now required -lcurand to solve `undefined symbol: curandCreateG…

…enerator` (#1879)
microsoft · Apr 5, 2022 · c487372 · c487372
1 parent 0149bd4
commit c487372
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/op_builder/cpu_adam.py b/op_builder/cpu_adam.py
@@ -18,6 +18,11 @@ def absolute_name(self):
     def sources(self):
         return ['csrc/adam/cpu_adam.cpp', 'csrc/common/custom_cuda_kernel.cu']
 
+    def libraries_args(self):
+        args = super().libraries_args()
+        args += ['curand']
+        return args
+
     def include_paths(self):
         import torch
         if not self.is_rocm_pytorch():