提交 0d60b591 编写于 作者: Q qijun

fix ci

上级 44bae42d
...@@ -41,7 +41,7 @@ class GetPlacesOp : public framework::OperatorBase { ...@@ -41,7 +41,7 @@ class GetPlacesOp : public framework::OperatorBase {
places.resize(trainer_count); places.resize(trainer_count);
if (device_type == "CUDA") { if (device_type == "CUDA") {
#ifdef PADDLE_WITH_CUDA #ifdef PADDLE_WITH_CUDA
PADDLE_ENFORCE_LT(trainer_count, GetCUDADeviceCount()); PADDLE_ENFORCE_LT(trainer_count, platform::GetCUDADeviceCount());
for (int i = 0; i < trainer_count; i++) { for (int i = 0; i < trainer_count; i++) {
places.emplace_back(platform::GPUPlace(i)); places.emplace_back(platform::GPUPlace(i));
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册