Skip to content

GitLab

Unverified Commit 6e5b5ca5 authored by Wentao Ye's avatar Wentao Ye Committed by GitHub
Browse files

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


Signed-off-by: default avataryewentao256 <zhyanwentao@126.com>
parent 488d8a98
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
csrc/moe/topk_softmax_kernels.cu
View file @ 6e5b5ca5
...@@ -20,6 +20,7 @@ ...@@ -20,6 +20,7 @@
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h> #include <c10/cuda/CUDAGuard.h>
#include "../cuda_compat.h" #include "../cuda_compat.h"
#include <cuda/std/functional>
#ifndef USE_ROCM #ifndef USE_ROCM
#include <cub/util_type.cuh> #include <cub/util_type.cuh>
...@@ -62,7 +63,7 @@ __launch_bounds__(TPB) __global__ ...@@ -62,7 +63,7 @@ __launch_bounds__(TPB) __global__
const int thread_row_offset = blockIdx.x * num_cols; const int thread_row_offset = blockIdx.x * num_cols;
cub::Sum sum; cuda::std::plus<float> sum;
float threadData(-FLT_MAX); float threadData(-FLT_MAX);
// Don't touch finished rows. // Don't touch finished rows.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment