Enable LTOIR to be linked from memory instead of files

rapidsai · Apr 18, 2024 · 4ef47fb · 4ef47fb
1 parent 1cef304
commit 4ef47fb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pynvjitlink/patch.py b/pynvjitlink/patch.py
@@ -98,6 +98,13 @@ class Object(LinkableCode):
     default_name = "<unnamed-object>"
 
 
+class LTOIR(LinkableCode):
+    """An LTOIR file in memory"""
+
+    kind = "ltoir"
+    default_name = "<unnamed-ltoir>"
+
+
 class PatchedLinker(Linker):
     def __init__(
         self,
@@ -259,3 +266,4 @@ def patch_numba_linker():
     cuda.Fatbin = Fatbin
     cuda.Object = Object
     cuda.PTXSource = PTXSource
+    cuda.LTOIR = LTOIR