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

merge uops with ops (#6111)

Browse Source 
Co-authored-by: chenyu <chenyu@fastmail.com>
This commit is contained in:
qazal
2024-08-17 06:17:57 +08:00
committed by GitHub
parent 379d080e74
commit 28c75bf2a6
41 changed files with 372 additions and 405 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_multitensor.py
View File

@@ -1,8 +1,7 @@
import unittest, functools, random
from typing import List
from tinygrad import Tensor, Device, nn, GlobalCounters, TinyJit, dtypes
from tinygrad.codegen.uops import UOps
from tinygrad.ops import MetaOps, ReduceOps, BinaryOps
from tinygrad.ops import MetaOps, ReduceOps, BinaryOps, UOps
from tinygrad.helpers import CI, getenv, prod, Context
from tinygrad.nn.state import get_parameters, get_state_dict
from tinygrad.engine.schedule import create_schedule