NVMATHPY-224 Linking LTOIR from memory instead of files

rapidsai · Apr 17, 2024 · 440f99a · 440f99a
1 parent 1cef304
commit 440f99a
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