github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-09 14:15:22 -05:00
Code Issues Packages Projects Releases Wiki Activity

cpu: add thread_id to worker (#11995)

Browse Source
This commit is contained in:
nimlgen
2025-09-04 14:58:13 +03:00
committed by GitHub
parent 35f37a64a9
commit e213b85810
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tinygrad/runtime/ops_cpu.py
View File

@@ -28,9 +28,9 @@ class ClangJITCompiler(Compiler):
def disassemble(self, lib:bytes): return capstone_flatdump(lib)
class CPUWorker(threading.Thread):
def __init__(self, dev):
def __init__(self, dev, tasks, thread_id):
super().__init__()
self.dev, self.tasks, self.daemon = dev, dev.tasks, True
self.dev, self.tasks, self.thread_id, self.daemon = dev, tasks, thread_id, True
def run(self):
while True:
@@ -121,5 +121,5 @@ class CPUAllocator(HCQAllocatorBase):
class CPUDevice(HCQCompiled):
def __init__(self, device:str=""):
self.tasks:queue.Queue = queue.Queue()
CPUWorker(self).start()
CPUWorker(self, self.tasks, thread_id=0).start()
super().__init__(device, CPUAllocator(self), ClangRenderer(), ClangJITCompiler(), functools.partial(CPUProgram, self), CPUSignal, CPUComputeQueue)

2
tinygrad/runtime/ops_llvm.py
View File

@@ -74,5 +74,5 @@ class HostLLVMCompiler(LLVMCompiler):
class LLVMDevice(HCQCompiled):
def __init__(self, device:str=""):
self.tasks:queue.Queue = queue.Queue()
CPUWorker(self).start()
CPUWorker(self, self.tasks, thread_id=0).start()
super().__init__(device, CPUAllocator(self), LLVMRenderer(), HostLLVMCompiler(), functools.partial(CPUProgram, self), HCQSignal, CPUComputeQueue)