diff --git a/paddle/fluid/operators/math/sequence_pooling.cu b/paddle/fluid/operators/math/sequence_pooling.cu index 124028ee1bcf2001091a7e3ff13dbbfa0850513e..422b06c70eb2107659666edf58223ae8e4666b1d 100644 --- a/paddle/fluid/operators/math/sequence_pooling.cu +++ b/paddle/fluid/operators/math/sequence_pooling.cu @@ -167,7 +167,7 @@ class SequencePoolFunctor { auto& lod = input.lod()[lod_level - 1]; const size_t item_dim = output->numel() / output->dims()[0]; dim3 threads(1024, 1); - dim3 grid(std::max(lod.size() - 1, 1UL), 1); + dim3 grid(std::max(static_cast(lod.size()) - 1, 1), 1); if (pooltype == "MAX") { sequence_pool_kernel< T, MaxPoolFunctor><<>>( @@ -331,7 +331,7 @@ class SequencePoolGradFunctor { auto& lod = in_grad->lod()[lod_level - 1]; const size_t item_dim = in_grad->numel() / in_grad->dims()[0]; dim3 threads(1024, 1); - dim3 grid(std::max(lod.size() - 1, 1UL), 1); + dim3 grid(std::max(static_cast(lod.size()) - 1, 1), 1); if (pooltype == "MAX") { sequence_pool_grad_kernel< T, MaxPoolGradFunctor><<>>(