Class Opt

tinygrad/codegen/opt/__init__.py:12–16 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

10
11	@dataclass(frozen=True, order=True)
12	class Opt:
13	op: OptOps
14	axis: int\|None = None
15	arg: int\|tuple\|None = None
16	def __repr__(self): return f"Opt(op={self.op}, axis={self.axis}, arg={self.arg})"
17
18	class KernelOptError(Exception): pass
19	def check(cond:bool, msg:str=""):

hand_coded_optimizationsFunction · 0.90

_apply_tc_optMethod · 0.90

search.pyFile · 0.90

get_kernel_actionsFunction · 0.90

tinygrad_nv_matmul.pyFile · 0.90

test_tc_upMethod · 0.90

test_gemm_tc_unrollMethod · 0.90

test_gemm_one_upcastedMethod · 0.90

test_gemm_upcastedMethod · 0.90

test_gemm_upcasted_localsMethod · 0.90

test_gemm_groupMethod · 0.90

test_reduce_groupMethod · 0.90

no outgoing calls

test_tc_upMethod · 0.72

test_gemm_tc_unrollMethod · 0.72

test_gemm_one_upcastedMethod · 0.72

test_gemm_upcastedMethod · 0.72

test_gemm_upcasted_localsMethod · 0.72

test_gemm_groupMethod · 0.72

test_reduce_groupMethod · 0.72

test_replay_with_optMethod · 0.72

test_reductionMethod · 0.72

test_arangeMethod · 0.72

test_upcast_0Method · 0.72

test_upcast_1Method · 0.72