зеркало из https://github.com/microsoft/DeepSpeed.git
add -lcurand to fix torch-nightly issue w. JIT (#1688)
This commit is contained in:
Родитель
3a4cb04243
Коммит
a85dce0728
|
@ -556,6 +556,9 @@ class CUDAOpBuilder(OpBuilder):
|
|||
|
||||
|
||||
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")
|
||||
|
|
Загрузка…
Ссылка в новой задаче