提交 ad3f6f4a 编写于 作者: W wanghaoshuang

Fix devices 'not undefined' error.

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