github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-07 13:15:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

remove _arg_int32 internal type (#2767)

Browse Source 
in DEFINE_GLOBAL, PtrDtype(int32) is buffer and int32 is int
This commit is contained in:
chenyu
2023-12-14 14:17:14 -05:00
committed by GitHub
parent 8a2a2257b4
commit 5235cdee3d
8 changed files with 26 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_dtype.py
View File

@@ -236,8 +236,7 @@ class TestTypeSpec(unittest.TestCase):
assert Tensor.ones([2,3,0]).sum(2).dtype == Tensor.default_type
# assert Tensor.ones([2,3,0], dtype=dtypes.int).sum(2).dtype == dtypes.int
# TODO: better way to write a set of core dtypes?
core_types = [d for d in DTYPES_DICT.values() if d not in [dtypes._arg_int32]]
core_types = list(DTYPES_DICT.values())
class TestTypePromotion(unittest.TestCase):
@given(st.sampled_from(core_types))
def test_self_promo_to_self(self, dtype):