nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

Revert "nv: check if jitlink is avail (#12731)" (#12735)

Browse Source 
This reverts commit a069a45d14.
This commit is contained in:
nimlgen
2025-10-16 20:41:49 +08:00
committed by GitHub
parent 8be7844b2e
commit cf9baeea61
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/support/compiler_cuda.py
+1 -3
View File
@@ -69,9 +69,7 @@ class PTXCompiler(Compiler):
def disassemble(self, lib:bytes): cuda_disassemble(lib, self.arch)
class NVPTXCompiler(PTXCompiler):
def __init__(self, arch:str):
nvrtc.nvJitLinkVersion(ctypes.byref(ctypes.c_int()), ctypes.byref(ctypes.c_int())) # try to get version to check if jitlink is available
super().__init__(arch, cache_key="nv_ptx")
def __init__(self, arch:str): super().__init__(arch, cache_key="nv_ptx")
def compile(self, src:str) -> bytes:
jitlink_check(nvrtc.nvJitLinkCreate(handle := nvrtc.nvJitLinkHandle(), 1, to_char_p_p([f'-arch={self.arch}'.encode()])), handle)
jitlink_check(nvrtc.nvJitLinkAddData(handle, nvrtc.NVJITLINK_INPUT_PTX, ptxsrc:=super().compile(src), len(ptxsrc), "<null>".encode()), handle)