From f037f85532dc270f3cd161c9f1504a09add754bd Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Mon, 23 Jun 2025 00:39:45 +0300 Subject: [PATCH] s/getenv("TC")/USE_TC context var (#10922) --- tinygrad/opt/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tinygrad/opt/__init__.py b/tinygrad/opt/__init__.py index b71ba9b982..efe6e96027 100644 --- a/tinygrad/opt/__init__.py +++ b/tinygrad/opt/__init__.py @@ -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)