add -lcurand to fix torch-nightly issue w. JIT (#1688)

microsoft · Jan 10, 2022 · a85dce0 · a85dce0
1 parent 3a4cb04
commit a85dce0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/op_builder/builder.py b/op_builder/builder.py
@@ -556,6 +556,9 @@ def libraries_args(self):
 
 
 class TorchCPUOpBuilder(CUDAOpBuilder):
+    def extra_ldflags(self):
+        return ['-lcurand']
+
     def cxx_args(self):
         import torch
         CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.CUDA_HOME, "lib64")