diff --git a/plugin/common/common.cuh b/plugin/common/common.cuh index d899416b..067e4e9f 100644 --- a/plugin/common/common.cuh +++ b/plugin/common/common.cuh @@ -326,7 +326,6 @@ template __device__ inline void scaledSoftmax( const int ld, const int lastValid, const float rsqrtHeadSize, const T* input, T* output) { - using BlockReduce = cub::BlockReduce; __shared__ typename BlockReduce::TempStorage tmpStorage; @@ -343,7 +342,6 @@ __device__ inline void scaledSoftmax( { threadData = 0; } - for (int i = threadIdx.x; i < lastValid; i += TPB) { const int idx = offset + i;