mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

devec 0

Browse source
This commit is contained in:
George Hotz 2025-03-25 17:12:43 +08:00
commit bce252e0b8
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tinygrad/codegen/devectorizer.py
View file

@ -158,6 +158,7 @@ def split_load_store(ctx:Renderer|None, ls:UOp, idx:UOp):
must_divide = True
if ctx is not None and ctx.device == "DSP":
lengths = [128,64,32,16,8,4]
if ls.dtype.count in [192]: return None # leave 192 alone
# we really want stores to be 128 for fast casting
#if ls.op is Ops.LOAD: lengths = [192]+lengths
#if ls.op is Ops.LOAD: lengths = [1536,1024,512,384,256,192,96]+lengths