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

bigint -> pyint [run_process_replay] (#6040)

Browse Source 
it's a python int. priority should be  higher than bool, but we are not using it in type promo now.
This commit is contained in:
chenyu
2024-08-12 09:12:23 -04:00
committed by GitHub
parent 059dd35985
commit 7ce716b3a0
4 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/helpers.py
View File

@@ -29,7 +29,7 @@ def assert_jit_cache_len(fxn, expected_len):
assert len(fxn.jit_cache[0].prg.jit_cache) == expected_len
def is_dtype_supported(dtype: DType, device: str = Device.DEFAULT):
if dtype == dtypes.bigint and device != "PYTHON": return False
if dtype == dtypes.pyint and device != "PYTHON": return False
if dtype == dtypes.bfloat16:
# NOTE: this requires bf16 buffer support
return device in {"AMD"} or (device in {"CUDA", "NV"} and not CI and not getenv("PTX"))