diff --git a/paddle/fluid/operators/random_crop_op.cu b/paddle/fluid/operators/random_crop_op.cu index 2782911b4fefc6f7cd37abc6df79a03803c8d07b..6fc9bedc55b4d349ddf3d109c7f9049113235f0c 100644 --- a/paddle/fluid/operators/random_crop_op.cu +++ b/paddle/fluid/operators/random_crop_op.cu @@ -14,6 +14,7 @@ #include "paddle/fluid/operators/random_crop_op.h" +namespace ops = paddle::operators; template using Kernel = ops::RandomCropKernel; REGISTER_OP_CUDA_KERNEL(random_crop, Kernel, Kernel, Kernel, diff --git a/paddle/fluid/operators/random_crop_op.h b/paddle/fluid/operators/random_crop_op.h index a34294f5eeb53c19b4089b76ffdf79716d227662..e0e24a7d1fa9f89a4cc528d6ccdd677d7944fbe8 100644 --- a/paddle/fluid/operators/random_crop_op.h +++ b/paddle/fluid/operators/random_crop_op.h @@ -60,7 +60,7 @@ HOSTDEVICE inline void StridedMemcpy(const T* x, const size_t* x_dims, T* out, size_t offset_i = offsets[i]; if (i == rank - 1) { - PADDLE_ENFORCE(x_stride == 1 && out_stride == 1); + PADDLE_ASSERT(x_stride == 1 && out_stride == 1); x += offset_i; for (size_t j = 0; j < out_dim_i; ++j) { *out++ = *x++; @@ -105,12 +105,12 @@ struct RandomCropFunctor { prod_batchsize_dims_ = 1; prod_x_ins_dims_ = 1; prod_out_ins_dims_ = 1; - for (size_t i = 0; i < rank_; ++i) { + for (size_t i = 0; i < static_cast(rank_); ++i) { size_t x_dim_i = x_dims[i]; size_t out_dim_i = out_dims[i]; x_dims_[i] = x_dim_i; out_dims_[i] = out_dim_i; - if (i < num_batchsize_dims_) { + if (i < static_cast(num_batchsize_dims_)) { PADDLE_ENFORCE_EQ(x_dim_i, out_dim_i); prod_batchsize_dims_ *= x_dim_i; } else {