mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-07-22 13:22:04 +08:00
Vehicle Researcher
161005a0c1
date: 2025-11-19T19:10:11 master commit: a29fdbd02407d41ecbcc69d151bb4837bfba3cbc
14 lines
572 B
Python
14 lines
572 B
Python
import unittest
|
|
from tinygrad import dtypes, Device
|
|
from tinygrad.device import is_dtype_supported
|
|
|
|
@unittest.skipUnless(Device.DEFAULT=="NULL", "Don't run when testing non-NULL backends")
|
|
class TestNULLSupportsDTypes(unittest.TestCase):
|
|
def test_null_supports_ints_floats_bool(self):
|
|
dts = dtypes.ints + dtypes.floats + (dtypes.bool,)
|
|
not_supported = [dt for dt in dts if not is_dtype_supported(dt, "NULL")]
|
|
self.assertFalse(not_supported, msg=f"expected these dtypes to be supported by NULL: {not_supported}")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|