[AMD] Fix Kimi-K3 SiTU JIT compile failure on ROCm

[RolaoDenthu]

Motivation

The Kimi-K3 SiTU activation kernels cannot be JIT-compiled on ROCm.situ_and_mul.cuh includes <cuda_fp8.h> unconditionally, which is a CUDA-only header, so hipcc fails with:

/sgl-workspace/sglang/python/sglang/kernels/jit/csrc/kimi_k3/situ_and_mul.cuh:21:10: fatal error: 'cuda_fp8.h' file not found
   21 | #include <cuda_fp8.h>
      |          ^~~~~~~~~~~~
1 error generated when compiling for gfx950.

Modifications

Guard the <cuda_fp8.h> include with #ifndef USE_ROCM

Accuracy Tests

Speed Tests and Profiling

Checklist

• Format your code according to the Format code with pre-commit.
• Add unit tests according to the Run and add unit tests.
• Update documentation according to Write documentations.
• Provide accuracy and speed benchmark results according to Test the accuracy and Benchmark the speed.
• Follow the SGLang code style guidance.

Review and Merge Process

1. Ping Merge Oncalls to start the process. See the PR Merge Process.
2. Get approvals from CODEOWNERS and other reviewers.
3. Trigger CI tests with comments or contact authorized users to do so.
   • Common commands include /tag-and-rerun-ci, /tag-run-ci-label, /rerun-failed-ci
4. After green CI and required approvals, ask Merge Oncalls or people with Write permission to merge the PR.

---

CI States

Latest PR Test (Base): ❌ Run #30865989327
Latest PR Test (Extra): ❌ Run #30865989193

[gemini-code-assist]

Caution

The consumer version of Gemini Code Assist on GitHub has been sunset. All code review activity has officially ceased.