github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

nv switch classlist_v2 to classlist (#5763)

Browse Source 
* nv switch classlist_v2 to classlist

* support in mockgpu

* fix mockgpu
This commit is contained in:
nimlgen
2024-07-28 20:24:42 +03:00
committed by GitHub
parent 73fda023d3
commit ea27ec4cd0
2 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/runtime/ops_nv.py
View File

@@ -396,8 +396,9 @@ class NVDevice(HCQCompiled):
return res_va
def _setup_nvclasses(self):
clsinfo = rmctrl.gpu_get_classlist_v2(self.fd_ctl, self.root, self.device)
self.nvclasses = {clsinfo.classList[i] for i in range(clsinfo.numClasses)}
classlist = memoryview(bytearray(100 * 4)).cast('I')
clsinfo = rmctrl.gpu_get_classlist(self.fd_ctl, self.root, self.device, numClasses=100, classList=mv_address(classlist))
self.nvclasses = {classlist[i] for i in range(clsinfo.numClasses)}
self.compute_class = next(clss for clss in [nv_gpu.ADA_COMPUTE_A, nv_gpu.AMPERE_COMPUTE_B] if clss in self.nvclasses)
def __init__(self, device:str=""):