Home
last modified time | relevance | path

Searched refs:gpu_rank_id (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/debug/
Dcommon.cc285 const std::string gpu_rank_id = common::GetEnv("OMPI_COMM_WORLD_RANK"); in GlogLogDirRegister() local
288 if (!rank_id.empty() && gpu_rank_id.empty()) { in GlogLogDirRegister()
290 } else if (rank_id.empty() && !gpu_rank_id.empty()) { in GlogLogDirRegister()
291 rank = std::string(gpu_rank_id); in GlogLogDirRegister()
292 } else if (!rank_id.empty() && !gpu_rank_id.empty()) { in GlogLogDirRegister()
/third_party/mindspore/mindspore/core/utils/
Dlog_adapter.cc466 std::string gpu_rank_id = mindspore::GetEnv("OMPI_COMM_WORLD_RANK"); local
468 if ((!rank_id.empty() && gpu_rank_id.empty()) || (!rank_id.empty() && !gpu_rank_id.empty())) {
470 } else if (rank_id.empty() && !gpu_rank_id.empty()) {
471 rank = gpu_rank_id;
/third_party/mindspore/mindspore/
Dlog.py487 gpu_rank_id = os.getenv('OMPI_COMM_WORLD_RANK')
489 if rank_id and gpu_rank_id:
494 elif gpu_rank_id:
495 rank = gpu_rank_id