mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

s/getenv("TC")/USE_TC context var (#10922)

Browse source
This commit is contained in:
qazal 2025-06-23 00:39:45 +03:00 • committed by GitHub
commit f037f85532
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tinygrad/opt/__init__.py
View file

@ -3,7 +3,7 @@
from tinygrad.opt.kernel import Kernel
from tinygrad.opt.heuristic import hand_coded_optimizations
from tinygrad.uop.ops import UOp
from tinygrad.helpers import NOOPT, BEAM, getenv
from tinygrad.helpers import NOOPT, BEAM, USE_TC, getenv
from tinygrad.renderer import Renderer
def get_optimized_ast(ast:UOp, renderer:Renderer) -> UOp:
@ -20,7 +20,7 @@ def get_optimized_ast(ast:UOp, renderer:Renderer) -> UOp:
k = Kernel(ast, opts=renderer)
if not NOOPT:
if not k.apply_tensor_cores(getenv("TC", 1)): k.apply_opts(hand_coded_optimizations(k))
if not k.apply_tensor_cores(USE_TC.value): k.apply_opts(hand_coded_optimizations(k))
if BEAM >= 1:
from tinygrad.opt.search import beam_search, bufs_from_lin
kb = Kernel(ast, opts=renderer)