github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-07 22:23:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

pickle

Browse Source
This commit is contained in:
George Hotz
2025-12-23 19:34:04 -05:00
parent 2621e57c53
commit 63447d50ef
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/renderer/ptx.py
View File

@@ -246,9 +246,11 @@ class CUDAPTXRenderer(PTXRenderer):
super().__init__(arch, "CUDA") super().__init__(arch, "CUDA")
from tinygrad.runtime.support.compiler_cuda import PTXCompiler from tinygrad.runtime.support.compiler_cuda import PTXCompiler
self.compiler = PTXCompiler(arch) self.compiler = PTXCompiler(arch)
def __reduce__(self): return self.__class__, (self.arch,)
class NVPTXRenderer(PTXRenderer): class NVPTXRenderer(PTXRenderer):
def __init__(self, arch:str): def __init__(self, arch:str):
super().__init__(arch, "NV") super().__init__(arch, "NV")
from tinygrad.runtime.support.compiler_cuda import NVPTXCompiler from tinygrad.runtime.support.compiler_cuda import NVPTXCompiler
self.compiler = NVPTXCompiler(arch) self.compiler = NVPTXCompiler(arch)
def __reduce__(self): return self.__class__, (self.arch,)