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

limit split to 1 due to windows path containing : (#944)

Browse Source
This commit is contained in:
Diogo
2023-06-06 13:27:54 -04:00
committed by GitHub
parent 079ea217a3
commit 3bb38c3518
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/lazy.py
View File

@@ -106,7 +106,7 @@ class LazyBuffer:
if GRAPH >= 3: log_op(self, self.op, phantom=True)
def __repr__(self): return f"<LB {self.shape} {self.dtype} op:{self.op.op if self.realized is None else self.realized} st:{self.st}>"
def _device_extra_args(self) -> Dict[str, str]: return {"device": self.device.split(":")[1]} if ":" in self.device else {}
def _device_extra_args(self) -> Dict[str, str]: return {"device": self.device.split(":", 1)[1]} if ":" in self.device else {}
def realize(self:LazyBuffer) -> LazyBuffer:
if self.realized is None: