merge uops with ops (#6111)

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

View File

@@ -1,13 +1,12 @@
import unittest
from typing import List, cast
import numpy as np
from tinygrad.codegen.uops import UOp, UOps
from tinygrad.device import Buffer, Device
from tinygrad.dtype import PtrDType, DType, dtypes
from tinygrad.engine.realize import CompiledRunner
from tinygrad.helpers import dedup, flatten
from tinygrad.renderer.cstyle import CStyleLanguage
from tinygrad.ops import BinaryOps
from tinygrad.ops import BinaryOps, UOp, UOps
from tinygrad.renderer import Program
from tinygrad.tensor import Tensor, _to_np_dtype
from tinygrad.lazy import LazyBuffer