fix thrust-transform-reduce error (#5078)

dce05da5 · Jun Gao · GitHub · 1cd7efc5 · dce05da5
Unverified Commit dce05da5 authored Nov 21, 2023 by Jun Gao Committed by GitHub Nov 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

colossalai/kernel/cuda_native/csrc/kernels/cuda_util.cu colossalai/kernel/cuda_native/csrc/kernels/cuda_util.cu +1 -1

No files found.
--- a/colossalai/kernel/cuda_native/csrc/kernels/cuda_util.cu
+++ b/colossalai/kernel/cuda_native/csrc/kernels/cuda_util.cu
 #include <thrust/device_vector.h>
 #include <thrust/reduce.h>
-
+#include <thrust/transform_reduce.h>
 #include "cuda_util.h"

 /* GPU function guard */