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

one less branch in Buffer.view [pr] (#13829)

Browse Source
This commit is contained in:
chenyu
2025-12-25 09:34:15 -05:00
committed by GitHub
parent 094753b4e0
commit 8985a4a023
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/device.py
View File

@@ -199,8 +199,7 @@ class Buffer:
return mv return mv
def view(self, size:int, dtype:DType, offset:int) -> Buffer: def view(self, size:int, dtype:DType, offset:int) -> Buffer:
assert offset < self.nbytes, "offset must be less than nbytes" assert offset < self.nbytes, "offset must be less than nbytes"
if self._base is not None: return Buffer(self.device, size, dtype, base=self._base, offset=self.offset+offset) return Buffer(self.device, size, dtype, base=self.base, offset=self.offset+offset)
return Buffer(self.device, size, dtype, base=self, offset=offset)
@dataclass(frozen=True) @dataclass(frozen=True)
class DMACPURef: class DMACPURef: