diff --git a/pynvjitlink/patch.py b/pynvjitlink/patch.py index 883d117..0e6a007 100644 --- a/pynvjitlink/patch.py +++ b/pynvjitlink/patch.py @@ -98,6 +98,13 @@ class Object(LinkableCode): default_name = "" +class LTOIR(LinkableCode): + """An LTOIR file in memory""" + + kind = 'ltoir' + default_name = "" + + 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