MCPcopy
hub / github.com/tinygrad/tinygrad / KernelInfo

Class KernelInfo

tinygrad/uop/ops.py:1038–1047  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1036
1037@dataclass(frozen=True)
1038class KernelInfo:
1039 name: str = "test" # name of the kernel
1040 axis_types: tuple[AxisType, ...] = tuple()
1041 dont_use_locals: bool = False # don't use local indexing
1042 applied_opts: tuple = tuple()
1043 opts_to_apply: tuple|None = None
1044 estimates: Estimates|None = None
1045 beam: int = 0
1046 @property
1047 def function_name(self): return to_function_name(self.name)
1048
1049@dataclass(frozen=True)
1050class ProgramInfo:

Callers 15

_zero_kernelFunction · 0.90
_embedding_bwd_kernelFunction · 0.90
dcache_flushFunction · 0.90
get_optimized_astMethod · 0.90
split_storeFunction · 0.90
canonicalize_astFunction · 0.90
hand_spec_kernel3Function · 0.90
amd_copy_matmulFunction · 0.90
hand_spec_tc_coresFunction · 0.90
amd_flash_attentionFunction · 0.90
asm_kernelFunction · 0.90

Calls

no outgoing calls

Tested by 15

test_global_prod_maxMethod · 0.72
fxnMethod · 0.72
test_reductionMethod · 0.72
test_arangeMethod · 0.72
test_kernel_infoMethod · 0.72
fxnMethod · 0.72
test_fail_1Method · 0.72
_setup_and_test_aluFunction · 0.72

Used in the wild real call sites across dependent graphs

searching dependent graphs…