"cpu only macro"

69b1b265 · dongzhihong · 9a16327b · 69b1b265
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/operators/random_op.h paddle/operators/random_op.h +2 -0

未找到文件。
--- a/paddle/operators/random_op.h
+++ b/paddle/operators/random_op.h
@@ -53,6 +53,7 @@ public:
          std,
          seed);
    } else {
+#ifndef PADDLE_ONLY_CPU
      Gaussian(
          dynamic_cast<platform::CUDADeviceContext*>(context.device_context_),
          output->mutable_data<T>(context.GetPlace()),
@@ -60,6 +61,7 @@ public:
          mean,
          std,
          seed);
+#endif
    }
  }
 };