зеркало из https://github.com/microsoft/DeepSpeed.git
fix "undefined symbol: curandCreateGenerator" for quantizer op (#3846)
This commit is contained in:
Родитель
c520d47679
Коммит
7e8bcc07d6
|
@ -623,7 +623,8 @@ class CUDAOpBuilder(OpBuilder):
|
|||
sources=self.strip_empty_entries(self.sources()),
|
||||
include_dirs=self.strip_empty_entries(self.include_paths()),
|
||||
libraries=self.strip_empty_entries(self.libraries_args()),
|
||||
extra_compile_args=compile_args)
|
||||
extra_compile_args=compile_args,
|
||||
extra_link_args=self.strip_empty_entries(self.extra_ldflags()))
|
||||
|
||||
if self.is_rocm_pytorch():
|
||||
# hip converts paths to absolute, this converts back to relative
|
||||
|
|
Загрузка…
Ссылка в новой задаче