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

test S_PACK in extra/assembly/amd/test/hw (#14537)

Browse Source 
* S_PACK_LL_B32_B16 in test/hw

* add rest of S_PACK instructions
This commit is contained in:
Christopher Milan
2026-02-04 11:17:16 -08:00
committed by GitHub
parent 9052db678f
commit 5338ce6b74
2 changed files with 44 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
extra/assembly/amd/test/hw/test_sop.py
View File

@@ -87,6 +87,50 @@ class TestBasicScalar(unittest.TestCase):
self.assertEqual(st.sgpr[2], f2i(2.0))
class TestPack(unittest.TestCase):
"""Tests for S_PACK instructions."""
def test_s_pack_ll_b32_b16(self):
"""S_PACK_LL_B32_B16 packs low 16 bits of two sources into one 32-bit result."""
instructions = [
s_mov_b32(s[0], 0xDEADAAAA),
s_mov_b32(s[1], 0xDEADBBBB),
s_pack_ll_b32_b16(s[2], s[0], s[1]),
]
st = run_program(instructions, n_lanes=1)
self.assertEqual(st.sgpr[2], 0xBBBBAAAA)
def test_s_pack_lh_b32_b16(self):
"""S_PACK_LH_B32_B16: D0 = { S1[31:16], S0[15:0] }."""
instructions = [
s_mov_b32(s[0], 0xDEADAAAA),
s_mov_b32(s[1], 0xDEADBBBB),
s_pack_lh_b32_b16(s[2], s[0], s[1]),
]
st = run_program(instructions, n_lanes=1)
self.assertEqual(st.sgpr[2], 0xDEADAAAA)
def test_s_pack_hh_b32_b16(self):
"""S_PACK_HH_B32_B16: D0 = { S1[31:16], S0[31:16] }."""
instructions = [
s_mov_b32(s[0], 0xDEADAAAA),
s_mov_b32(s[1], 0xDEADBBBB),
s_pack_hh_b32_b16(s[2], s[0], s[1]),
]
st = run_program(instructions, n_lanes=1)
self.assertEqual(st.sgpr[2], 0xDEADDEAD)
def test_s_pack_hl_b32_b16(self):
"""S_PACK_HL_B32_B16: D0 = { S1[15:0], S0[31:16] }."""
instructions = [
s_mov_b32(s[0], 0xDEADAAAA),
s_mov_b32(s[1], 0xDEADBBBB),
s_pack_hl_b32_b16(s[2], s[0], s[1]),
]
st = run_program(instructions, n_lanes=1)
self.assertEqual(st.sgpr[2], 0xBBBBDEAD)
class TestQuadmaskWqm(unittest.TestCase):
"""Tests for S_QUADMASK_B32 and S_WQM_B32."""

10
extra/assembly/amd/test/test_emu2_pcode.py
View File

@@ -62,16 +62,6 @@ class TestWithSources(unittest.TestCase):
dest, val = assigns[0]
self.assertEqual(val.op, Ops.MUL)
def test_s_pack_ll_b32_b16(self):
"""Test S_PACK_LL_B32_B16 packs two 16-bit values into 32-bit result."""
s0 = UOp.const(dtypes.uint32, 0xDEADAAAA)
s1 = UOp.const(dtypes.uint32, 0xDEADBBBB)
_, assigns = parse_pcode(PCODE[SOP2Op.S_PACK_LL_B32_B16], {'S0': s0, 'S1': s1})
self.assertEqual(len(assigns), 1)
dest, val = assigns[0]
self.assertTrue(dest.startswith('D0'))
self.assertEqual(val.simplify().arg, 0xBBBBAAAA)
class TestParseExpr(unittest.TestCase):
"""Test the parse_expr function directly."""