Support mrope triton kernel and add unit test (#11722)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com> Co-authored-by: b8zhong <b8zhong@uwaterloo.ca>
2025-10-20 11:51:07 +08:00
parent c4e81e64fb
commit 271d3d0d50
4 changed files with 630 additions and 1 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -77,6 +77,7 @@ suites = {
        TestFile("test_eval_fp8_accuracy.py", 303),
        TestFile("test_fa3.py", 376),
        # TestFile("test_flashmla.py", 352),
+        TestFile("rotary_embedding/test_mrope.py", 300),
        TestFile("test_function_call_parser.py", 10),
        TestFile("test_fused_moe.py", 30),
        TestFile("test_gpt_oss_1gpu.py", 600),