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

Cleanup vectorized hip renders (#2497)

Browse Source 
* add typedefs and make_dtypen functions

use ext_vector_type for half16 kernels

* remove the old test_render because we just use whatever cstyle has

* align vectors
This commit is contained in:
qazal
2023-11-29 17:02:12 -05:00
committed by GitHub
parent abfc99187d
commit 370cfbb957
2 changed files with 5 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/test_renderer.py
View File

@@ -1,11 +0,0 @@
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}")