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

move exec_alu from uops to ops (#4033)

Browse Source 
will use this for const folding in lazy too
This commit is contained in:
chenyu
2024-04-01 20:20:53 -04:00
committed by GitHub
parent 82440d3416
commit 0a34d6016b
3 changed files with 34 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/runtime/ops_python.py
View File

@@ -6,8 +6,8 @@ import pickle, base64, itertools, time, struct
from tinygrad.dtype import DType, dtypes, ImageDType
from tinygrad.helpers import all_same, getenv, flatten
from tinygrad.device import Compiled, Allocator, Compiler, CompilerOptions
from tinygrad.codegen.uops import UOpGraph, UOps, exec_alu
from tinygrad.ops import BinaryOps, TernaryOps
from tinygrad.codegen.uops import UOpGraph, UOps
from tinygrad.ops import BinaryOps, TernaryOps, exec_alu
def _load(m, i):
if i < 0 or i >= len(m): raise IndexError(f"load out of bounds, size is {len(m)} and access is {i}")