[ROCm] [Bugfix] Fix fused_qknorm_rope_kernel rocm compatibility (#28500)

Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
2025-11-12 05:01:14 -08:00
parent c5f10cc139
commit edb59a9470
6 changed files with 37 additions and 38 deletions
--- a/tests/kernels/core/test_fused_qk_norm_rope.py
+++ b/tests/kernels/core/test_fused_qk_norm_rope.py
@@ -44,8 +44,8 @@ def _apply_qk_norm_rope(


@pytest.mark.skipif(
-    not current_platform.is_cuda(),
-    reason="fused_qk_norm_rope custom op requires cuda platform",
+    not current_platform.is_cuda_alike(),
+    reason="fused_qk_norm_rope custom op requires cuda and rocm platform",
 )
@pytest.mark.parametrize("device", CUDA_DEVICES)
@pytest.mark.parametrize("dtype", DTYPES)