From 13f5d39fcf6ee89beaf43dbd2f3455db48da6f85 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 2 Jun 2026 13:16:42 -0700 Subject: [PATCH] count --- tinygrad/renderer/isa/x86.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tinygrad/renderer/isa/x86.py b/tinygrad/renderer/isa/x86.py index 5fe2ac1029..34e0a4b567 100644 --- a/tinygrad/renderer/isa/x86.py +++ b/tinygrad/renderer/isa/x86.py @@ -633,8 +633,8 @@ def encode(x:UOp, opc:int, reg:int|None=None, pp:int=0, sel:int=0, we:int=0) -> reg = cast(int, cast(Register, reg_uop.reg).index if reg_uop is not None else reg) rm = cast(Register, rm_uop.reg).index idx = cast(Register, idx_uop.reg).index if idx_uop is not None and idx_uop.reg is not None else 4 - rm_sz = 8 if isinstance(rm_uop.dtype, PtrDType) and disp_uop is None else rm_uop.dtype.itemsize - reg_sz = (reg_uop.dtype.itemsize if not isinstance(reg_uop.dtype, PtrDType) else 8) if reg_uop is not None else 0 + rm_sz = 8 if isinstance(rm_uop.dtype, PtrDType) and disp_uop is None else (rm_uop.dtype.itemsize*rm_uop.dtype.count) + reg_sz = ((reg_uop.dtype.itemsize*reg_uop.dtype.count) if not isinstance(reg_uop.dtype, PtrDType) else 8) if reg_uop is not None else 0 sz = reg_sz or rm_sz # encode instruction