Fix debug build

ProExpertProg · ProExpertProg · commit 8a049cfae484 · 2025-06-18T10:26:19.000-04:00
Signed-off-by: Luka Govedic &lt;lgovedic@redhat.com&gt;
diff --git a/csrc/moe/topk_softmax_kernels.cu b/csrc/moe/topk_softmax_kernels.cu
@@ -530,7 +530,7 @@ void topk_softmax(
             stream);
     }
     else {
-        assert(topk_indices.scalar_type() == at::ScalarType::Int64);
+        TORCH_CHECK(topk_indices.scalar_type() == at::ScalarType::Long);
         vllm::moe::topkGatingSoftmaxKernelLauncher(
             gating_output.data_ptr<float>(),
             topk_weights.data_ptr<float>(),

Original file line number	Diff line number	Diff line change
`@@ -530,7 +530,7 @@ void topk_softmax(`
`530`	`530`	`stream);`
`531`	`531`	`}`
`532`	`532`	`else {`
`533`		`- assert(topk_indices.scalar_type() == at::ScalarType::Int64);`
	`533`	`+ TORCH_CHECK(topk_indices.scalar_type() == at::ScalarType::Long);`
`534`	`534`	`vllm::moe::topkGatingSoftmaxKernelLauncher(`
`535`	`535`	`gating_output.data_ptr<float>(),`
`536`	`536`	`topk_weights.data_ptr<float>(),`