提交 ad3f6f4a 编写于 作者: W wanghaoshuang

Fix devices 'not undefined' error.

上级 3d96b381
......@@ -69,9 +69,9 @@ void InitP2P(std::vector<int> devices) {
}
void InitDevices(bool init_p2p) {
/*Init all available devices by default */
#ifdef PADDLE_WITH_CUDA
/*Init all available devices by default */
std::vector<int> devices;
#ifdef PADDLE_WITH_CUDA
try {
int count = platform::GetCUDADeviceCount();
for (int i = 0; i < count; ++i) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册