fix(rocm): Enable non-gated MoE (is_act_and_mul=False) support on ROCm

[rabi]

Purpose

Models like NemotronH use non-gated MoE with activations like relu2_no_mul. Previously, this was blocked on ROCm because the platform check only allowed CUDA.

• Updates platform check from is_cuda() to is_cuda_alike() to allow ROCm
• Disables AITER kernel for non-gated MoE since AITER only supports gated activations (silu/gelu)
• Falls back to Triton implementation which properly handles non-gated activations via apply_moe_activation()

Test Plan

Tested on AMD MI210 GPU with NemotronH model.

Test Result

Model loads and serves successfully.

[gemini-code-assist]

Code Review

This pull request correctly enables non-gated MoE support on ROCm. The changes are well-targeted and logical. You've correctly updated the platform check from is_cuda() to is_cuda_alike() to include ROCm. Additionally, you've properly disabled the AITER kernel for non-gated MoE, as it only supports gated activations, allowing the system to fall back to the Triton implementation which handles this case. The changes appear correct and align with the stated purpose. I have no further comments.

[tjtanaa]

LGTM

[mergify]

This pull request has merge conflicts that must be resolved before it can be
merged. Please rebase the PR, @rabi.

https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

[mergify]

Hi @rabi, the pre-commit checks have failed. Please run:

uv pip install pre-commit
pre-commit install
pre-commit run --all-files

Then, commit the changes and push to your branch.

For future commits, pre-commit will run automatically on changed files before each commit.

Tip

Is mypy or markdownlint failing?

mypy and markdownlint are run differently in CI. If the failure is related to either of these checks, please use the following commands to run them locally:

# For mypy (substitute "3.10" with the failing version if needed)
pre-commit run --hook-stage manual mypy-3.10
# For markdownlint
pre-commit run --hook-stage manual markdownlint

[tjtanaa]

Thank you for the fix.