From abb4d30476d1da13c8548030277ab289ae2ada2b Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 22 Oct 2025 19:54:41 +0800 Subject: [PATCH] hotfix: slightly looser load spec for AMD bfloat16 --- tinygrad/uop/spec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/uop/spec.py b/tinygrad/uop/spec.py index 71a1379768..0e16188cf1 100644 --- a/tinygrad/uop/spec.py +++ b/tinygrad/uop/spec.py @@ -114,7 +114,7 @@ program_spec = PatternMatcher([ (UPat(Ops.INDEX, src=(UPat(GroupOp.Defines).or_after(), UPat())), lambda: True), # LOAD (idx, alt_value) / LOAD(idx) / STORE(idx, val) - (UPat(Ops.LOAD, src=(UPat(Ops.INDEX, name="idx").or_casted(), UPat((Ops.VECTORIZE, Ops.VCONST, Ops.CONST)))), validate_index), + (UPat(Ops.LOAD, src=(UPat(Ops.INDEX, name="idx").or_casted(), UPat())), validate_index), (UPat(Ops.LOAD, src=(UPat(Ops.INDEX, name="idx").or_casted(), )), validate_index), (UPat(Ops.STORE, src=(UPat(Ops.INDEX, name="idx").or_casted(), UPat())), validate_index),