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

[CPU] refactor rope kernels (#31897)

Browse Source 
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Ma Mingfei
2026-07-22 09:12:02 +08:00
committed by GitHub
co-authored by Copilot Autofix powered by AI
parent 14c0a31829
commit 878d77929d
3 changed files with 373 additions and 866 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/cpu/test_rope.py
-1
View File
@@ -161,7 +161,6 @@ class TestROPE(CustomTestCase):
atol = rtol = precision[q_pe.dtype]
torch.testing.assert_close(q_pe, q_pe_clone, atol=atol, rtol=rtol)
torch.testing.assert_close(k_pe, k_pe_clone, atol=atol, rtol=rtol)
torch.testing.assert_close(k_pe, k_pe_clone)
def test_origin_rope(self):
def single_test(