From 397a2e6eb6cff0d956b22c46f63e109694695391 Mon Sep 17 00:00:00 2001 From: chenyu Date: Tue, 12 Nov 2024 14:52:14 -0500 Subject: [PATCH] no special case for int32 in truncate [pr] (#7657) this masked an issue that idx is not data, and should never need truncate --- tinygrad/dtype.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tinygrad/dtype.py b/tinygrad/dtype.py index 564fb30c8e..6f2217148f 100644 --- a/tinygrad/dtype.py +++ b/tinygrad/dtype.py @@ -184,5 +184,5 @@ truncate: Dict[DType, Callable] = {dtypes.bool: bool, dtypes.float16: truncate_fp16, dtypes.float32: lambda x: ctypes.c_float(x).value, dtypes.float64: lambda x: ctypes.c_double(x).value, dtypes.uint8: lambda x: ctypes.c_uint8(x).value, dtypes.uint16: lambda x: ctypes.c_uint16(x).value, dtypes.uint32: lambda x: ctypes.c_uint32(x).value, dtypes.uint64: lambda x: ctypes.c_uint64(x).value, - dtypes.int8: lambda x: ctypes.c_int8(x).value, dtypes.int16: lambda x: ctypes.c_int16(x).value, dtypes.int32: lambda x: ctypes.c_int32(x).value \ - if isinstance(x,int) else x, dtypes.int64: lambda x: ctypes.c_int64(x).value} + dtypes.int8: lambda x: ctypes.c_int8(x).value, dtypes.int16: lambda x: ctypes.c_int16(x).value, dtypes.int32: lambda x: ctypes.c_int32(x).value, + dtypes.int64: lambda x: ctypes.c_int64(x).value}