follow comments

Author: zhiqiu

paddle/pten/kernels/bernoulli_kernel.h

@@ -19,19 +19,6 @@
 
 namespace pten {
 
-template <typename T>
-inline HOSTDEVICE T BernoulliFunctor(T p, T rand) {
- PADDLE_ENFORCE_LE(p,
- 1.0,
- pten::errors::OutOfRange(
- "The probability should be <= 1, but got %f", p));
- PADDLE_ENFORCE_GE(p,
- 0.0,
- pten::errors::OutOfRange(
- "The probability should be >= 0, but got %f", p));
- return static_cast<T>(rand < p);
-}
-
 template <typename T, typename Context>
 void BernoulliKernel(const Context& ctx,
  const DenseTensor& x,

paddle/pten/kernels/cpu/bernoulli_kernel.cc

@@ -19,6 +19,19 @@
 
 namespace pten {
 
+template <typename T>
+inline T BernoulliFunctor(T p, T rand) {
+ PADDLE_ENFORCE_LE(p,
+ 1.0,
+ pten::errors::OutOfRange(
+ "The probability should be <= 1, but got %f", p));
+ PADDLE_ENFORCE_GE(p,
+ 0.0,
+ pten::errors::OutOfRange(
+ "The probability should be >= 0, but got %f", p));
+ return static_cast<T>(rand < p);
+}
+
 template <typename T, typename Context>
 void BernoulliKernel(const Context& ctx,
  const DenseTensor& x,