mirror of https://github.com/hpcaitech/ColossalAI
fix thrust-transform-reduce error (#5078)
parent
1cd7efc520
commit
dce05da535
|
@ -1,6 +1,6 @@
|
|||
#include <thrust/device_vector.h>
|
||||
#include <thrust/reduce.h>
|
||||
|
||||
#include <thrust/transform_reduce.h>
|
||||
#include "cuda_util.h"
|
||||
|
||||
/* GPU function guard */
|
||||
|
|
Loading…
Reference in New Issue