提交 69b1b265 编写于 作者: D dongzhihong

"cpu only macro"

上级 9a16327b
...@@ -53,6 +53,7 @@ public: ...@@ -53,6 +53,7 @@ public:
std, std,
seed); seed);
} else { } else {
#ifndef PADDLE_ONLY_CPU
Gaussian( Gaussian(
dynamic_cast<platform::CUDADeviceContext*>(context.device_context_), dynamic_cast<platform::CUDADeviceContext*>(context.device_context_),
output->mutable_data<T>(context.GetPlace()), output->mutable_data<T>(context.GetPlace()),
...@@ -60,6 +61,7 @@ public: ...@@ -60,6 +61,7 @@ public:
mean, mean,
std, std,
seed); seed);
#endif
} }
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册