From 4017bbdacb864183fc9d863a8ffce77700290956 Mon Sep 17 00:00:00 2001
From: Mitch Lewis <mitch.lewis930@gmail.com>
Date: Tue, 13 Jan 2026 18:29:46 -0700
Subject: [PATCH] Revert "[BugFix] Fix topk_softmax assert (#19764)"

This reverts commit 4f35be10a96feeca0328d3ab8d359e1eaae5c23d.
---
 csrc/moe/topk_softmax_kernels.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/csrc/moe/topk_softmax_kernels.cu b/csrc/moe/topk_softmax_kernels.cu
index cd80bfda7dfd..99c52ef17d08 100644
--- a/csrc/moe/topk_softmax_kernels.cu
+++ b/csrc/moe/topk_softmax_kernels.cu
@@ -573,7 +573,7 @@ void topk_softmax(
             stream);
     }
     else {
-        TORCH_CHECK(topk_indices.scalar_type() == at::ScalarType::Long);
+        assert(topk_indices.scalar_type() == at::ScalarType::Int64);
         vllm::moe::topkGatingSoftmaxKernelLauncher(
             gating_output.data_ptr<float>(),
             topk_weights.data_ptr<float>(),