biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Refactor] Fix Compile Warning #1444-D (#21208)

Browse Source 
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Wentao Ye
2025-07-22 02:33:51 -04:00
committed by GitHub
parent 488d8a986a
commit 6e5b5ca580
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
csrc/moe/topk_softmax_kernels.cu
View File

@@ -20,6 +20,7 @@
#include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h>
#include "../cuda_compat.h"
#include <cuda/std/functional>
#ifndef USE_ROCM
#include <cub/util_type.cuh>
@@ -62,7 +63,7 @@ __launch_bounds__(TPB) __global__
const int thread_row_offset = blockIdx.x * num_cols;
cub::Sum sum;
cuda::std::plus<float> sum;
float threadData(-FLT_MAX);
// Don't touch finished rows.