github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-19 02:44:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

devectorize prereqs [pr] (#9404)

Browse Source
This commit is contained in:
George Hotz
2025-03-11 12:33:29 +08:00
committed by GitHub
parent beed00eabe
commit 2780e2027e
3 changed files with 7 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/dtype.py
View File

@@ -54,6 +54,8 @@ class PtrDType(DType):
def vec(self, sz:int) -> DType:
assert self.v == 1, f"can't vectorize ptr {self} with size {sz}"
if sz == 1: return self # sz=1 is a scalar
if isinstance(self, ImageDType):
return ImageDType(self.priority, self.itemsize, self.name, self.fmt, self.count, self, self._base, self.local, sz, self.size, self.shape)
return type(self)(self.priority, self.itemsize, self.name, self.fmt, self.count, self, self._base, self.local, sz, self.size)
def ptr(self, size=-1, local=False): raise RuntimeError("can't make a pointer from a pointer")
@property