minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[AMD] Fix DSA device-to-host direct test on rocm720 (page_size%16 assert) (#27850)

Browse Source
This commit is contained in:
Michael
2026-06-11 14:24:23 -07:00
committed by GitHub
parent df5055e00f
commit 12d3f02be8
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/unit/mem_cache/test_dsa_pool_host_unit.py
+6
View File
@@ -153,6 +153,12 @@ class TestDSAHiCacheTransfer(unittest.TestCase):
def test_device_to_host_indexer_kernel(self): def test_device_to_host_indexer_kernel(self):
self._run_device_to_host_indexer_copy(io_backend="kernel") self._run_device_to_host_indexer_copy(io_backend="kernel")
@unittest.skipIf(
is_hip(),
"DSATokenToKVPool with page_size=1 (used on HIP) trips the ROCm 7.2.0 "
"HIP-preshuffle assert `page_size % 16 == 0` during pool construction "
"(seen on pr-test-amd-rocm720). The rest of this file passes on AMD.",
)
def test_device_to_host_indexer_direct(self): def test_device_to_host_indexer_direct(self):
self._run_device_to_host_indexer_copy(io_backend="direct") self._run_device_to_host_indexer_copy(io_backend="direct")