github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

use_tensor_cores is a heuristic (#11989)

Browse Source 
* use_tensor_cores is a heuristic

* context
This commit is contained in:
George Hotz
2025-09-03 17:05:10 -07:00
committed by GitHub
parent 63e930fec3
commit a5f2b4872a
5 changed files with 42 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_linearizer.py
View File

@@ -616,7 +616,8 @@ class TestLinearizer(unittest.TestCase):
"""
x, y = Tensor.randn(64,64), Tensor.randn(64,64)
out = x.matmul(y)
k = helper_linearizer_opt(out)[-1]
with Context(TC=0):
k = helper_linearizer_opt(out)[-1]
uops = get_program(k.ast, k.opts, k.applied_opts).uops
# check that the float4 cast collapses
store_vals = [u.src[1] for u in uops if u.op is Ops.STORE and u.src[0].dtype.addrspace != AddrSpace.REG]