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

don't exclude weakint in is_dtype_supported [pr] (#15378)

Browse Source
This commit is contained in:
chenyu
2026-03-20 02:08:29 -04:00
committed by GitHub
parent cf6a429aaa
commit 459ef41ea0
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tinygrad/device.py
View File

@@ -331,7 +331,6 @@ class Compiled:
# TODO: move this to each Device
# this only tracks if the dtype is natively supported, it may be supported in the frontend using decomps
def is_dtype_supported(dtype:DType, device:str|None=None) -> bool:
if dtype == dtypes.weakint: return False
if device is None: device = Device.DEFAULT
if dtype == dtypes.bfloat16:
if device == "METAL": return not CI