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

bring back pyint (#7150)

Browse Source 
fixed test_failure_52 and resnet. need to understand this better
This commit is contained in:
chenyu
2024-10-18 14:54:37 -04:00
committed by GitHub
parent c8b59416d0
commit 37e83ca6fc
8 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/dtype.py
View File

@@ -84,6 +84,7 @@ class dtypes:
@staticmethod
def fields() -> Dict[str, DType]: return DTYPES_DICT
void: Final[DType] = DType(-1, 0, "void", None, 1)
pyint: Final[DType] = DType(-1, 8, "pyint", None, 1) # arbitrary precision integer, same itemsize to int64 so min/max works
bool: Final[DType] = DType(0, 1, "bool", '?', 1)
int8: Final[DType] = DType(1, 1, "char", 'b', 1)
uint8: Final[DType] = DType(2, 1, "unsigned char", 'B', 1)
@@ -115,7 +116,7 @@ class dtypes:
floats = (float16, bfloat16, float32, float64)
uints = (uint8, uint16, uint32, uint64)
sints = (int8, int16, int32, int64)
sints = (int8, int16, int32, int64, pyint)
ints = uints + sints
if (env_default_float := getenv("DEFAULT_FLOAT", "")):