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

move is_dtype_supported to device [pr] (#7575)

Browse Source
This commit is contained in:
George Hotz
2024-11-07 20:38:03 +08:00
committed by GitHub
parent 1f5ea1e412
commit 205befa788
23 changed files with 50 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_dtype.py
View File

@@ -2,12 +2,13 @@ import unittest, operator, subprocess, math
import numpy as np
import torch
from typing import Any, List
from tinygrad.device import is_dtype_supported
from tinygrad.helpers import getenv, DEBUG, CI
from tinygrad.dtype import DType, DTYPES_DICT, ImageDType, PtrDType, least_upper_float, least_upper_dtype, truncate_fp16
from tinygrad import Device, Tensor, dtypes
from tinygrad.tensor import _to_np_dtype
from hypothesis import given, settings, strategies as strat
from test.helpers import is_dtype_supported, rand_for_dtype
from test.helpers import rand_for_dtype
import pytest
pytestmark = pytest.mark.filterwarnings("ignore")