diff --git a/paddle/pten/kernels/functions/general/elementwise_functor.h b/paddle/pten/kernels/functions/general/elementwise_functor.h index 9218726c266e0..f0d4305ea6cd9 100644 --- a/paddle/pten/kernels/functions/general/elementwise_functor.h +++ b/paddle/pten/kernels/functions/general/elementwise_functor.h @@ -123,6 +123,20 @@ struct SameDimsDivFunctor { DenseTensor* z); }; +template +struct SameDimsDivFunctor< + DevCtx, + T, + typename std::enable_if::value>::type> { + void operator()(const DevCtx& dev_ctx, + const DenseTensor& x, + const DenseTensor& y, + DenseTensor* z) { + paddle::platform::errors::InvalidArgument( + "If use SameDimsDivFunctor, template args(T) must be floating point. "); + } +}; + template struct SameDimsDivFunctor< DevCtx,