|
|
|
@ -19,8 +19,7 @@
|
|
|
|
|
#include <cstdlib>
|
|
|
|
|
#include <fstream>
|
|
|
|
|
#include "utils/ms_utils.h"
|
|
|
|
|
#include "minddata/dataset/core/config_manager.h"
|
|
|
|
|
#include "minddata/dataset/core/global_context.h"
|
|
|
|
|
#include "minddata/dataset/util/path.h"
|
|
|
|
|
#include "minddata/dataset/engine/perf/monitor.h"
|
|
|
|
|
#include "minddata/dataset/engine/perf/device_queue_tracing.h"
|
|
|
|
|
#include "minddata/dataset/engine/perf/connector_size.h"
|
|
|
|
@ -28,7 +27,6 @@
|
|
|
|
|
#include "minddata/dataset/engine/perf/cpu_sampling.h"
|
|
|
|
|
#include "minddata/dataset/engine/perf/dataset_iterator_tracing.h"
|
|
|
|
|
#include "minddata/dataset/util/log_adapter.h"
|
|
|
|
|
#include "minddata/dataset/util/path.h"
|
|
|
|
|
|
|
|
|
|
namespace mindspore {
|
|
|
|
|
namespace dataset {
|
|
|
|
@ -61,13 +59,11 @@ Status ProfilingManager::Initialize() {
|
|
|
|
|
#endif
|
|
|
|
|
dir_path_ = real_path;
|
|
|
|
|
|
|
|
|
|
std::shared_ptr<ConfigManager> cfg = GlobalContext::config_manager();
|
|
|
|
|
int32_t rank_id = cfg->rank_id();
|
|
|
|
|
// If DEVICE_ID is not set, default value is 0
|
|
|
|
|
if (rank_id < 0) {
|
|
|
|
|
rank_id = 0;
|
|
|
|
|
device_id_ = common::GetEnv("DEVICE_ID");
|
|
|
|
|
if (device_id_.empty()) {
|
|
|
|
|
device_id_ = "0";
|
|
|
|
|
}
|
|
|
|
|
device_id_ = std::to_string(rank_id);
|
|
|
|
|
|
|
|
|
|
// Register all profiling node.
|
|
|
|
|
// device_queue node is used for graph mode
|
|
|
|
|