mirror of https://github.com/commaai/tinygrad.git
12 lines
760 B
Python
12 lines
760 B
Python
import unittest
|
|
from tinygrad.helpers import dtypes
|
|
|
|
from tinygrad.renderer.hip import HIPLanguage
|
|
|
|
class TestRenderer(unittest.TestCase):
|
|
def test_render_cast(self):
|
|
self.assertEqual(HIPLanguage().render_cast(["data0"], dtypes.half), "(half)(data0)")
|
|
self.assertEqual(HIPLanguage().render_cast(["data0", "data1", "data2", "data3"], dtypes.float.vec(4)), "make_float4(data0,data1,data2,data3)")
|
|
self.assertEqual(HIPLanguage().render_cast(["data0", "data1", "data2", "data3", "data4", "data5", "data6", "data7"], dtypes.float.vec(8)), "{data0,data1,data2,data3,data4,data5,data6,data7}")
|
|
self.assertEqual(HIPLanguage().render_cast(["data0", "data1", "data2", "data3"], dtypes.half.vec(4)), "{(half)data0,(half)data1,(half)data2,(half)data3}")
|