Use add_library to link archives in Numba patch

rapidsai · Feb 13, 2024 · 34afabc · 34afabc
1 parent 90bc3f2
commit 34afabc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pynvjitlink/patch.py b/pynvjitlink/patch.py
@@ -104,7 +104,7 @@ def add_file(self, path, kind):
         elif kind == FILE_EXTENSION_MAP["fatbin"]:
             fn = self._linker.add_fatbin
         elif kind == FILE_EXTENSION_MAP["a"]:
-            raise LinkerError("Don't know how to link archives")
+            fn = self._linker.add_library
         elif kind == FILE_EXTENSION_MAP["ptx"]:
             return self.add_ptx(data, name)
         elif kind == FILE_EXTENSION_MAP["o"]: