tinygrad 0.9.1__py3-none-any.whl → 0.10.0__py3-none-any.whl

tinygrad/runtime/autogen/nvrtc.py

@@ -0,0 +1,579 @@
+# mypy: ignore-errors
+# -*- coding: utf-8 -*-
+#
+# TARGET arch is: []
+# WORD_SIZE is: 8
+# POINTER_SIZE is: 8
+# LONGDOUBLE_SIZE is: 16
+#
+import ctypes, ctypes.util
+
+
+_libraries = {}
+_libraries['libnvrtc.so'] = ctypes.CDLL(ctypes.util.find_library('nvrtc'))
+def string_cast(char_pointer, encoding='utf-8', errors='strict'):
+    value = ctypes.cast(char_pointer, ctypes.c_char_p).value
+    if value is not None and encoding is not None:
+        value = value.decode(encoding, errors=errors)
+    return value
+
+
+def char_pointer_cast(string, encoding='utf-8'):
+    if encoding is not None:
+        try:
+            string = string.encode(encoding)
+        except AttributeError:
+            # In Python3, bytes has no encode attribute
+            pass
+    string = ctypes.c_char_p(string)
+    return ctypes.cast(string, ctypes.POINTER(ctypes.c_char))
+
+
+
+class AsDictMixin:
+    @classmethod
+    def as_dict(cls, self):
+        result = {}
+        if not isinstance(self, AsDictMixin):
+            # not a structure, assume it's already a python object
+            return self
+        if not hasattr(cls, "_fields_"):
+            return result
+        # sys.version_info >= (3, 5)
+        # for (field, *_) in cls._fields_:  # noqa
+        for field_tuple in cls._fields_:  # noqa
+            field = field_tuple[0]
+            if field.startswith('PADDING_'):
+                continue
+            value = getattr(self, field)
+            type_ = type(value)
+            if hasattr(value, "_length_") and hasattr(value, "_type_"):
+                # array
+                if not hasattr(type_, "as_dict"):
+                    value = [v for v in value]
+                else:
+                    type_ = type_._type_
+                    value = [type_.as_dict(v) for v in value]
+            elif hasattr(value, "contents") and hasattr(value, "_type_"):
+                # pointer
+                try:
+                    if not hasattr(type_, "as_dict"):
+                        value = value.contents
+                    else:
+                        type_ = type_._type_
+                        value = type_.as_dict(value.contents)
+                except ValueError:
+                    # nullptr
+                    value = None
+            elif isinstance(value, AsDictMixin):
+                # other structure
+                value = type_.as_dict(value)
+            result[field] = value
+        return result
+
+
+class Structure(ctypes.Structure, AsDictMixin):
+
+    def __init__(self, *args, **kwds):
+        # We don't want to use positional arguments fill PADDING_* fields
+
+        args = dict(zip(self.__class__._field_names_(), args))
+        args.update(kwds)
+        super(Structure, self).__init__(**args)
+
+    @classmethod
+    def _field_names_(cls):
+        if hasattr(cls, '_fields_'):
+            return (f[0] for f in cls._fields_ if not f[0].startswith('PADDING'))
+        else:
+            return ()
+
+    @classmethod
+    def get_type(cls, field):
+        for f in cls._fields_:
+            if f[0] == field:
+                return f[1]
+        return None
+
+    @classmethod
+    def bind(cls, bound_fields):
+        fields = {}
+        for name, type_ in cls._fields_:
+            if hasattr(type_, "restype"):
+                if name in bound_fields:
+                    if bound_fields[name] is None:
+                        fields[name] = type_()
+                    else:
+                        # use a closure to capture the callback from the loop scope
+                        fields[name] = (
+                            type_((lambda callback: lambda *args: callback(*args))(
+                                bound_fields[name]))
+                        )
+                    del bound_fields[name]
+                else:
+                    # default callback implementation (does nothing)
+                    try:
+                        default_ = type_(0).restype().value
+                    except TypeError:
+                        default_ = None
+                    fields[name] = type_((
+                        lambda default_: lambda *args: default_)(default_))
+            else:
+                # not a callback function, use default initialization
+                if name in bound_fields:
+                    fields[name] = bound_fields[name]
+                    del bound_fields[name]
+                else:
+                    fields[name] = type_()
+        if len(bound_fields) != 0:
+            raise ValueError(
+                "Cannot bind the following unknown callback(s) {}.{}".format(
+                    cls.__name__, bound_fields.keys()
+            ))
+        return cls(**fields)
+
+
+class Union(ctypes.Union, AsDictMixin):
+    pass
+
+
+
+_libraries['libnvJitLink.so'] = ctypes.CDLL(ctypes.util.find_library('nvJitLink'))
+c_int128 = ctypes.c_ubyte*16
+c_uint128 = c_int128
+void = None
+if ctypes.sizeof(ctypes.c_longdouble) == 16:
+    c_long_double_t = ctypes.c_longdouble
+else:
+    c_long_double_t = ctypes.c_ubyte*16
+
+
+
+
+# values for enumeration 'c__EA_nvrtcResult'
+c__EA_nvrtcResult__enumvalues = {
+    0: 'NVRTC_SUCCESS',
+    1: 'NVRTC_ERROR_OUT_OF_MEMORY',
+    2: 'NVRTC_ERROR_PROGRAM_CREATION_FAILURE',
+    3: 'NVRTC_ERROR_INVALID_INPUT',
+    4: 'NVRTC_ERROR_INVALID_PROGRAM',
+    5: 'NVRTC_ERROR_INVALID_OPTION',
+    6: 'NVRTC_ERROR_COMPILATION',
+    7: 'NVRTC_ERROR_BUILTIN_OPERATION_FAILURE',
+    8: 'NVRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION',
+    9: 'NVRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION',
+    10: 'NVRTC_ERROR_NAME_EXPRESSION_NOT_VALID',
+    11: 'NVRTC_ERROR_INTERNAL_ERROR',
+    12: 'NVRTC_ERROR_TIME_FILE_WRITE_FAILED',
+}
+NVRTC_SUCCESS = 0
+NVRTC_ERROR_OUT_OF_MEMORY = 1
+NVRTC_ERROR_PROGRAM_CREATION_FAILURE = 2
+NVRTC_ERROR_INVALID_INPUT = 3
+NVRTC_ERROR_INVALID_PROGRAM = 4
+NVRTC_ERROR_INVALID_OPTION = 5
+NVRTC_ERROR_COMPILATION = 6
+NVRTC_ERROR_BUILTIN_OPERATION_FAILURE = 7
+NVRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION = 8
+NVRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION = 9
+NVRTC_ERROR_NAME_EXPRESSION_NOT_VALID = 10
+NVRTC_ERROR_INTERNAL_ERROR = 11
+NVRTC_ERROR_TIME_FILE_WRITE_FAILED = 12
+c__EA_nvrtcResult = ctypes.c_uint32 # enum
+nvrtcResult = c__EA_nvrtcResult
+nvrtcResult__enumvalues = c__EA_nvrtcResult__enumvalues
+try:
+    nvrtcGetErrorString = _libraries['libnvrtc.so'].nvrtcGetErrorString
+    nvrtcGetErrorString.restype = ctypes.POINTER(ctypes.c_char)
+    nvrtcGetErrorString.argtypes = [nvrtcResult]
+except AttributeError:
+    pass
+try:
+    nvrtcVersion = _libraries['libnvrtc.so'].nvrtcVersion
+    nvrtcVersion.restype = nvrtcResult
+    nvrtcVersion.argtypes = [ctypes.POINTER(ctypes.c_int32), ctypes.POINTER(ctypes.c_int32)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetNumSupportedArchs = _libraries['libnvrtc.so'].nvrtcGetNumSupportedArchs
+    nvrtcGetNumSupportedArchs.restype = nvrtcResult
+    nvrtcGetNumSupportedArchs.argtypes = [ctypes.POINTER(ctypes.c_int32)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetSupportedArchs = _libraries['libnvrtc.so'].nvrtcGetSupportedArchs
+    nvrtcGetSupportedArchs.restype = nvrtcResult
+    nvrtcGetSupportedArchs.argtypes = [ctypes.POINTER(ctypes.c_int32)]
+except AttributeError:
+    pass
+class struct__nvrtcProgram(Structure):
+    pass
+
+nvrtcProgram = ctypes.POINTER(struct__nvrtcProgram)
+try:
+    nvrtcCreateProgram = _libraries['libnvrtc.so'].nvrtcCreateProgram
+    nvrtcCreateProgram.restype = nvrtcResult
+    nvrtcCreateProgram.argtypes = [ctypes.POINTER(ctypes.POINTER(struct__nvrtcProgram)), ctypes.POINTER(ctypes.c_char), ctypes.POINTER(ctypes.c_char), ctypes.c_int32, ctypes.POINTER(ctypes.POINTER(ctypes.c_char)), ctypes.POINTER(ctypes.POINTER(ctypes.c_char))]
+except AttributeError:
+    pass
+try:
+    nvrtcDestroyProgram = _libraries['libnvrtc.so'].nvrtcDestroyProgram
+    nvrtcDestroyProgram.restype = nvrtcResult
+    nvrtcDestroyProgram.argtypes = [ctypes.POINTER(ctypes.POINTER(struct__nvrtcProgram))]
+except AttributeError:
+    pass
+try:
+    nvrtcCompileProgram = _libraries['libnvrtc.so'].nvrtcCompileProgram
+    nvrtcCompileProgram.restype = nvrtcResult
+    nvrtcCompileProgram.argtypes = [nvrtcProgram, ctypes.c_int32, ctypes.POINTER(ctypes.POINTER(ctypes.c_char))]
+except AttributeError:
+    pass
+try:
+    nvrtcGetPTXSize = _libraries['libnvrtc.so'].nvrtcGetPTXSize
+    nvrtcGetPTXSize.restype = nvrtcResult
+    nvrtcGetPTXSize.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetPTX = _libraries['libnvrtc.so'].nvrtcGetPTX
+    nvrtcGetPTX.restype = nvrtcResult
+    nvrtcGetPTX.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetCUBINSize = _libraries['libnvrtc.so'].nvrtcGetCUBINSize
+    nvrtcGetCUBINSize.restype = nvrtcResult
+    nvrtcGetCUBINSize.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetCUBIN = _libraries['libnvrtc.so'].nvrtcGetCUBIN
+    nvrtcGetCUBIN.restype = nvrtcResult
+    nvrtcGetCUBIN.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetNVVMSize = _libraries['libnvrtc.so'].nvrtcGetNVVMSize
+    nvrtcGetNVVMSize.restype = nvrtcResult
+    nvrtcGetNVVMSize.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetNVVM = _libraries['libnvrtc.so'].nvrtcGetNVVM
+    nvrtcGetNVVM.restype = nvrtcResult
+    nvrtcGetNVVM.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetLTOIRSize = _libraries['libnvrtc.so'].nvrtcGetLTOIRSize
+    nvrtcGetLTOIRSize.restype = nvrtcResult
+    nvrtcGetLTOIRSize.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetLTOIR = _libraries['libnvrtc.so'].nvrtcGetLTOIR
+    nvrtcGetLTOIR.restype = nvrtcResult
+    nvrtcGetLTOIR.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetOptiXIRSize = _libraries['libnvrtc.so'].nvrtcGetOptiXIRSize
+    nvrtcGetOptiXIRSize.restype = nvrtcResult
+    nvrtcGetOptiXIRSize.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetOptiXIR = _libraries['libnvrtc.so'].nvrtcGetOptiXIR
+    nvrtcGetOptiXIR.restype = nvrtcResult
+    nvrtcGetOptiXIR.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetProgramLogSize = _libraries['libnvrtc.so'].nvrtcGetProgramLogSize
+    nvrtcGetProgramLogSize.restype = nvrtcResult
+    nvrtcGetProgramLogSize.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetProgramLog = _libraries['libnvrtc.so'].nvrtcGetProgramLog
+    nvrtcGetProgramLog.restype = nvrtcResult
+    nvrtcGetProgramLog.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvrtcAddNameExpression = _libraries['libnvrtc.so'].nvrtcAddNameExpression
+    nvrtcAddNameExpression.restype = nvrtcResult
+    nvrtcAddNameExpression.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvrtcGetLoweredName = _libraries['libnvrtc.so'].nvrtcGetLoweredName
+    nvrtcGetLoweredName.restype = nvrtcResult
+    nvrtcGetLoweredName.argtypes = [nvrtcProgram, ctypes.POINTER(ctypes.c_char), ctypes.POINTER(ctypes.POINTER(ctypes.c_char))]
+except AttributeError:
+    pass
+
+# values for enumeration 'c__EA_nvJitLinkResult'
+c__EA_nvJitLinkResult__enumvalues = {
+    0: 'NVJITLINK_SUCCESS',
+    1: 'NVJITLINK_ERROR_UNRECOGNIZED_OPTION',
+    2: 'NVJITLINK_ERROR_MISSING_ARCH',
+    3: 'NVJITLINK_ERROR_INVALID_INPUT',
+    4: 'NVJITLINK_ERROR_PTX_COMPILE',
+    5: 'NVJITLINK_ERROR_NVVM_COMPILE',
+    6: 'NVJITLINK_ERROR_INTERNAL',
+    7: 'NVJITLINK_ERROR_THREADPOOL',
+    8: 'NVJITLINK_ERROR_UNRECOGNIZED_INPUT',
+}
+NVJITLINK_SUCCESS = 0
+NVJITLINK_ERROR_UNRECOGNIZED_OPTION = 1
+NVJITLINK_ERROR_MISSING_ARCH = 2
+NVJITLINK_ERROR_INVALID_INPUT = 3
+NVJITLINK_ERROR_PTX_COMPILE = 4
+NVJITLINK_ERROR_NVVM_COMPILE = 5
+NVJITLINK_ERROR_INTERNAL = 6
+NVJITLINK_ERROR_THREADPOOL = 7
+NVJITLINK_ERROR_UNRECOGNIZED_INPUT = 8
+c__EA_nvJitLinkResult = ctypes.c_uint32 # enum
+nvJitLinkResult = c__EA_nvJitLinkResult
+nvJitLinkResult__enumvalues = c__EA_nvJitLinkResult__enumvalues
+
+# values for enumeration 'c__EA_nvJitLinkInputType'
+c__EA_nvJitLinkInputType__enumvalues = {
+    0: 'NVJITLINK_INPUT_NONE',
+    1: 'NVJITLINK_INPUT_CUBIN',
+    2: 'NVJITLINK_INPUT_PTX',
+    3: 'NVJITLINK_INPUT_LTOIR',
+    4: 'NVJITLINK_INPUT_FATBIN',
+    5: 'NVJITLINK_INPUT_OBJECT',
+    6: 'NVJITLINK_INPUT_LIBRARY',
+    10: 'NVJITLINK_INPUT_ANY',
+}
+NVJITLINK_INPUT_NONE = 0
+NVJITLINK_INPUT_CUBIN = 1
+NVJITLINK_INPUT_PTX = 2
+NVJITLINK_INPUT_LTOIR = 3
+NVJITLINK_INPUT_FATBIN = 4
+NVJITLINK_INPUT_OBJECT = 5
+NVJITLINK_INPUT_LIBRARY = 6
+NVJITLINK_INPUT_ANY = 10
+c__EA_nvJitLinkInputType = ctypes.c_uint32 # enum
+nvJitLinkInputType = c__EA_nvJitLinkInputType
+nvJitLinkInputType__enumvalues = c__EA_nvJitLinkInputType__enumvalues
+class struct_nvJitLink(Structure):
+    pass
+
+nvJitLinkHandle = ctypes.POINTER(struct_nvJitLink)
+uint32_t = ctypes.c_uint32
+try:
+    __nvJitLinkCreate_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkCreate_12_4
+    __nvJitLinkCreate_12_4.restype = nvJitLinkResult
+    __nvJitLinkCreate_12_4.argtypes = [ctypes.POINTER(ctypes.POINTER(struct_nvJitLink)), uint32_t, ctypes.POINTER(ctypes.POINTER(ctypes.c_char))]
+except AttributeError:
+    pass
+try:
+    nvJitLinkCreate = _libraries['libnvJitLink.so'].nvJitLinkCreate
+    nvJitLinkCreate.restype = nvJitLinkResult
+    nvJitLinkCreate.argtypes = [ctypes.POINTER(ctypes.POINTER(struct_nvJitLink)), uint32_t, ctypes.POINTER(ctypes.POINTER(ctypes.c_char))]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkDestroy_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkDestroy_12_4
+    __nvJitLinkDestroy_12_4.restype = nvJitLinkResult
+    __nvJitLinkDestroy_12_4.argtypes = [ctypes.POINTER(ctypes.POINTER(struct_nvJitLink))]
+except AttributeError:
+    pass
+try:
+    nvJitLinkDestroy = _libraries['libnvJitLink.so'].nvJitLinkDestroy
+    nvJitLinkDestroy.restype = nvJitLinkResult
+    nvJitLinkDestroy.argtypes = [ctypes.POINTER(ctypes.POINTER(struct_nvJitLink))]
+except AttributeError:
+    pass
+size_t = ctypes.c_uint64
+try:
+    __nvJitLinkAddData_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkAddData_12_4
+    __nvJitLinkAddData_12_4.restype = nvJitLinkResult
+    __nvJitLinkAddData_12_4.argtypes = [nvJitLinkHandle, nvJitLinkInputType, ctypes.POINTER(None), size_t, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkAddData = _libraries['libnvJitLink.so'].nvJitLinkAddData
+    nvJitLinkAddData.restype = nvJitLinkResult
+    nvJitLinkAddData.argtypes = [nvJitLinkHandle, nvJitLinkInputType, ctypes.POINTER(None), size_t, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkAddFile_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkAddFile_12_4
+    __nvJitLinkAddFile_12_4.restype = nvJitLinkResult
+    __nvJitLinkAddFile_12_4.argtypes = [nvJitLinkHandle, nvJitLinkInputType, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkAddFile = _libraries['libnvJitLink.so'].nvJitLinkAddFile
+    nvJitLinkAddFile.restype = nvJitLinkResult
+    nvJitLinkAddFile.argtypes = [nvJitLinkHandle, nvJitLinkInputType, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkComplete_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkComplete_12_4
+    __nvJitLinkComplete_12_4.restype = nvJitLinkResult
+    __nvJitLinkComplete_12_4.argtypes = [nvJitLinkHandle]
+except AttributeError:
+    pass
+try:
+    nvJitLinkComplete = _libraries['libnvJitLink.so'].nvJitLinkComplete
+    nvJitLinkComplete.restype = nvJitLinkResult
+    nvJitLinkComplete.argtypes = [nvJitLinkHandle]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkGetLinkedCubinSize_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkGetLinkedCubinSize_12_4
+    __nvJitLinkGetLinkedCubinSize_12_4.restype = nvJitLinkResult
+    __nvJitLinkGetLinkedCubinSize_12_4.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkGetLinkedCubinSize = _libraries['libnvJitLink.so'].nvJitLinkGetLinkedCubinSize
+    nvJitLinkGetLinkedCubinSize.restype = nvJitLinkResult
+    nvJitLinkGetLinkedCubinSize.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkGetLinkedCubin_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkGetLinkedCubin_12_4
+    __nvJitLinkGetLinkedCubin_12_4.restype = nvJitLinkResult
+    __nvJitLinkGetLinkedCubin_12_4.argtypes = [nvJitLinkHandle, ctypes.POINTER(None)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkGetLinkedCubin = _libraries['libnvJitLink.so'].nvJitLinkGetLinkedCubin
+    nvJitLinkGetLinkedCubin.restype = nvJitLinkResult
+    nvJitLinkGetLinkedCubin.argtypes = [nvJitLinkHandle, ctypes.POINTER(None)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkGetLinkedPtxSize_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkGetLinkedPtxSize_12_4
+    __nvJitLinkGetLinkedPtxSize_12_4.restype = nvJitLinkResult
+    __nvJitLinkGetLinkedPtxSize_12_4.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkGetLinkedPtxSize = _libraries['libnvJitLink.so'].nvJitLinkGetLinkedPtxSize
+    nvJitLinkGetLinkedPtxSize.restype = nvJitLinkResult
+    nvJitLinkGetLinkedPtxSize.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkGetLinkedPtx_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkGetLinkedPtx_12_4
+    __nvJitLinkGetLinkedPtx_12_4.restype = nvJitLinkResult
+    __nvJitLinkGetLinkedPtx_12_4.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkGetLinkedPtx = _libraries['libnvJitLink.so'].nvJitLinkGetLinkedPtx
+    nvJitLinkGetLinkedPtx.restype = nvJitLinkResult
+    nvJitLinkGetLinkedPtx.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkGetErrorLogSize_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkGetErrorLogSize_12_4
+    __nvJitLinkGetErrorLogSize_12_4.restype = nvJitLinkResult
+    __nvJitLinkGetErrorLogSize_12_4.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkGetErrorLogSize = _libraries['libnvJitLink.so'].nvJitLinkGetErrorLogSize
+    nvJitLinkGetErrorLogSize.restype = nvJitLinkResult
+    nvJitLinkGetErrorLogSize.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkGetErrorLog_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkGetErrorLog_12_4
+    __nvJitLinkGetErrorLog_12_4.restype = nvJitLinkResult
+    __nvJitLinkGetErrorLog_12_4.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkGetErrorLog = _libraries['libnvJitLink.so'].nvJitLinkGetErrorLog
+    nvJitLinkGetErrorLog.restype = nvJitLinkResult
+    nvJitLinkGetErrorLog.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkGetInfoLogSize_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkGetInfoLogSize_12_4
+    __nvJitLinkGetInfoLogSize_12_4.restype = nvJitLinkResult
+    __nvJitLinkGetInfoLogSize_12_4.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkGetInfoLogSize = _libraries['libnvJitLink.so'].nvJitLinkGetInfoLogSize
+    nvJitLinkGetInfoLogSize.restype = nvJitLinkResult
+    nvJitLinkGetInfoLogSize.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_uint64)]
+except AttributeError:
+    pass
+try:
+    __nvJitLinkGetInfoLog_12_4 = _libraries['libnvJitLink.so'].__nvJitLinkGetInfoLog_12_4
+    __nvJitLinkGetInfoLog_12_4.restype = nvJitLinkResult
+    __nvJitLinkGetInfoLog_12_4.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkGetInfoLog = _libraries['libnvJitLink.so'].nvJitLinkGetInfoLog
+    nvJitLinkGetInfoLog.restype = nvJitLinkResult
+    nvJitLinkGetInfoLog.argtypes = [nvJitLinkHandle, ctypes.POINTER(ctypes.c_char)]
+except AttributeError:
+    pass
+try:
+    nvJitLinkVersion = _libraries['libnvJitLink.so'].nvJitLinkVersion
+    nvJitLinkVersion.restype = nvJitLinkResult
+    nvJitLinkVersion.argtypes = [ctypes.POINTER(ctypes.c_uint32), ctypes.POINTER(ctypes.c_uint32)]
+except AttributeError:
+    pass
+__all__ = \
+    ['NVJITLINK_ERROR_INTERNAL', 'NVJITLINK_ERROR_INVALID_INPUT',
+    'NVJITLINK_ERROR_MISSING_ARCH', 'NVJITLINK_ERROR_NVVM_COMPILE',
+    'NVJITLINK_ERROR_PTX_COMPILE', 'NVJITLINK_ERROR_THREADPOOL',
+    'NVJITLINK_ERROR_UNRECOGNIZED_INPUT',
+    'NVJITLINK_ERROR_UNRECOGNIZED_OPTION', 'NVJITLINK_INPUT_ANY',
+    'NVJITLINK_INPUT_CUBIN', 'NVJITLINK_INPUT_FATBIN',
+    'NVJITLINK_INPUT_LIBRARY', 'NVJITLINK_INPUT_LTOIR',
+    'NVJITLINK_INPUT_NONE', 'NVJITLINK_INPUT_OBJECT',
+    'NVJITLINK_INPUT_PTX', 'NVJITLINK_SUCCESS',
+    'NVRTC_ERROR_BUILTIN_OPERATION_FAILURE',
+    'NVRTC_ERROR_COMPILATION', 'NVRTC_ERROR_INTERNAL_ERROR',
+    'NVRTC_ERROR_INVALID_INPUT', 'NVRTC_ERROR_INVALID_OPTION',
+    'NVRTC_ERROR_INVALID_PROGRAM',
+    'NVRTC_ERROR_NAME_EXPRESSION_NOT_VALID',
+    'NVRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION',
+    'NVRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION',
+    'NVRTC_ERROR_OUT_OF_MEMORY',
+    'NVRTC_ERROR_PROGRAM_CREATION_FAILURE',
+    'NVRTC_ERROR_TIME_FILE_WRITE_FAILED', 'NVRTC_SUCCESS',
+    '__nvJitLinkAddData_12_4', '__nvJitLinkAddFile_12_4',
+    '__nvJitLinkComplete_12_4', '__nvJitLinkCreate_12_4',
+    '__nvJitLinkDestroy_12_4', '__nvJitLinkGetErrorLogSize_12_4',
+    '__nvJitLinkGetErrorLog_12_4', '__nvJitLinkGetInfoLogSize_12_4',
+    '__nvJitLinkGetInfoLog_12_4',
+    '__nvJitLinkGetLinkedCubinSize_12_4',
+    '__nvJitLinkGetLinkedCubin_12_4',
+    '__nvJitLinkGetLinkedPtxSize_12_4',
+    '__nvJitLinkGetLinkedPtx_12_4', 'c__EA_nvJitLinkInputType',
+    'c__EA_nvJitLinkResult', 'c__EA_nvrtcResult', 'nvJitLinkAddData',
+    'nvJitLinkAddFile', 'nvJitLinkComplete', 'nvJitLinkCreate',
+    'nvJitLinkDestroy', 'nvJitLinkGetErrorLog',
+    'nvJitLinkGetErrorLogSize', 'nvJitLinkGetInfoLog',
+    'nvJitLinkGetInfoLogSize', 'nvJitLinkGetLinkedCubin',
+    'nvJitLinkGetLinkedCubinSize', 'nvJitLinkGetLinkedPtx',
+    'nvJitLinkGetLinkedPtxSize', 'nvJitLinkHandle',
+    'nvJitLinkInputType', 'nvJitLinkInputType__enumvalues',
+    'nvJitLinkResult', 'nvJitLinkResult__enumvalues',
+    'nvJitLinkVersion', 'nvrtcAddNameExpression',
+    'nvrtcCompileProgram', 'nvrtcCreateProgram',
+    'nvrtcDestroyProgram', 'nvrtcGetCUBIN', 'nvrtcGetCUBINSize',
+    'nvrtcGetErrorString', 'nvrtcGetLTOIR', 'nvrtcGetLTOIRSize',
+    'nvrtcGetLoweredName', 'nvrtcGetNVVM', 'nvrtcGetNVVMSize',
+    'nvrtcGetNumSupportedArchs', 'nvrtcGetOptiXIR',
+    'nvrtcGetOptiXIRSize', 'nvrtcGetPTX', 'nvrtcGetPTXSize',
+    'nvrtcGetProgramLog', 'nvrtcGetProgramLogSize',
+    'nvrtcGetSupportedArchs', 'nvrtcProgram', 'nvrtcResult',
+    'nvrtcResult__enumvalues', 'nvrtcVersion', 'size_t',
+    'struct__nvrtcProgram', 'struct_nvJitLink', 'uint32_t']

tinygrad/runtime/autogen/opencl.py

@@ -610,17 +610,17 @@ CL_KHRONOS_VENDOR_ID_CODEPLAY = 0x10004 # macro
 CL_VERSION_MAJOR_BITS = (10) # macro
 CL_VERSION_MINOR_BITS = (10) # macro
 CL_VERSION_PATCH_BITS = (12) # macro
-# CL_VERSION_MAJOR_MASK = ((1<<(10)) # macro
-# CL_VERSION_MINOR_MASK = ((1<<(10)) # macro
-# CL_VERSION_PATCH_MASK = ((1<<(12)) # macro
-# def CL_VERSION_MAJOR(version):  # macro
-#    return ((version)>>((10)+(12)))
-# def CL_VERSION_MINOR(version):  # macro
-#    return (((version)>>(12))&((1<<(10)))
-# def CL_VERSION_PATCH(version):  # macro
-#    return ((version)&((1<<(12)))
-# def CL_MAKE_VERSION(major, minor, patch):  # macro
-#    return ((((major)&((1<<(10)))<<((10)+(12)))|(((minor)&((1<<(10)))<<(12))|((patch)&((1<<(12))))
+CL_VERSION_MAJOR_MASK = ((1<<(10))-1) # macro
+CL_VERSION_MINOR_MASK = ((1<<(10))-1) # macro
+CL_VERSION_PATCH_MASK = ((1<<(12))-1) # macro
+def CL_VERSION_MAJOR(version):  # macro
+   return ((version)>>((10)+(12)))
+def CL_VERSION_MINOR(version):  # macro
+   return (((version)>>(12))&((1<<(10))-1))
+def CL_VERSION_PATCH(version):  # macro
+   return ((version)&((1<<(12))-1))
+def CL_MAKE_VERSION(major, minor, patch):  # macro
+   return ((((major)&((1<<(10))-1))<<((10)+(12)))|(((minor)&((1<<(10))-1))<<(12))|((patch)&((1<<(12))-1)))
 class struct__cl_platform_id(Structure):
     pass