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

7e8bcc07 · Jinzhen Lin · GitHub · c520d476 · 7e8bcc07
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

op_builder/builder.py op_builder/builder.py +2 -1

未找到文件。
--- a/op_builder/builder.py
+++ b/op_builder/builder.py
@@ -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