github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

bitcast

Browse Source
This commit is contained in:
George Hotz
2026-01-11 09:50:19 +09:00
parent 18fabb8723
commit 322eb1fbc8
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
extra/assembly/amd/pcode_transform.py
View File

@@ -264,9 +264,13 @@ pcode_spec = PatternMatcher([
# ASSIGN: dtype matches rhs (unless both void)
(UPat(Ops.ASSIGN, src=(UPat.var("lhs"), UPat.var("rhs")), name="a"),
lambda a, lhs, rhs: a.dtype == rhs.dtype and (rhs.dtype != dtypes.void or lhs.dtype == dtypes.void)),
# Pcode-specific ops (void sources allowed - type comes from context)
# BITCAST: void source allowed (type view on untyped register)
(UPat(Ops.BITCAST, src=(UPat(),)), lambda: True),
(UPat((Ops.CUSTOMI, Ops.CUSTOM, Ops.CAT)), lambda: True),
# CUSTOMI/CAT: must be typed (slice bounds or bit concat determine type)
(UPat(Ops.CUSTOMI, name="x"), lambda x: x.dtype != dtypes.void),
(UPat(Ops.CAT, name="x"), lambda x: x.dtype != dtypes.void),
# CUSTOM: MEM and passthrough ops (abs, cvtToQuietNAN) can be void (wrapped by BITCAST/CAST)
(UPat(Ops.CUSTOM, name="x"), lambda x: x.dtype != dtypes.void or x.arg in {'MEM', 'abs', 'cvtToQuietNAN'}),
# POW allows int exponent with float base
(UPat(Ops.POW, dtype=dtypes.floats, src=(UPat(dtype=dtypes.floats), UPat(dtype=dtypes.ints))), lambda: True),
]) + shared_spec