fix "undefined symbol: curandCreateGenerator" for quantizer op (#3846)

This commit is contained in:
Jinzhen Lin 2023-07-08 03:36:02 +08:00 коммит произвёл GitHub
Родитель c520d47679
Коммит 7e8bcc07d6
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 2 добавлений и 1 удалений

Просмотреть файл

@ -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