Fix typo of selected_npus (#31230)

revert-31562-mean
Leo Chen 4 years ago committed by GitHub
parent 387c1db4f1
commit d45f5d787e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -77,7 +77,7 @@ int GetCurrentNPUDeviceId() {
std::vector<int> GetSelectedNPUDevices() {
// use user specified NPUs in single-node multi-process mode.
std::vector<int> devices;
if (!FLAGS_selected_gpus.empty()) {
if (!FLAGS_selected_npus.empty()) {
auto devices_str = paddle::string::Split(FLAGS_selected_npus, ',');
for (auto id : devices_str) {
devices.push_back(atoi(id.c_str()));

@ -64,7 +64,7 @@ int main(int argc, char** argv) {
undefok.push_back("initial_cpu_memory_in_mb");
#endif
#if defined(PADDLE_WITH_CUDA)
#if defined(PADDLE_WITH_ASCEND_CL)
envs.push_back("selected_npus");
#endif

Loading…
Cancel
Save