Browse Source

add trigger point for distribute task failed

1. add trigger point
2. reduce unuse import
3. modify GetRealFilePath
tags/v1.2.0-rc1
luopengting 4 years ago
parent
commit
db5de7fd28
3 changed files with 7 additions and 5 deletions
  1. +1
    -3
      mindspore/ccsrc/debug/rdr/base_recorder.h
  2. +0
    -2
      mindspore/ccsrc/debug/rdr/recorder_manager.cc
  3. +6
    -0
      mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc

+ 1
- 3
mindspore/ccsrc/debug/rdr/base_recorder.h View File

@@ -30,9 +30,7 @@ class BaseRecorder {
public:
BaseRecorder() : module_(""), tag_(""), directory_(""), filename_(""), timestamp_("") {}
BaseRecorder(const std::string &module, const std::string &tag) : module_(module), tag_(tag), filename_("") {
auto &config_parser_ptr = mindspore::EnvConfigParser::GetInstance();
config_parser_ptr.Parse();
directory_ = config_parser_ptr.rdr_path();
directory_ = mindspore::EnvConfigParser::GetInstance().rdr_path();

if (tag.length() > maxTagLength) {
tag_ = tag.substr(0, maxTagLength);


+ 0
- 2
mindspore/ccsrc/debug/rdr/recorder_manager.cc View File

@@ -17,8 +17,6 @@
#include <utility>
#include "debug/rdr/base_recorder.h"
#include "debug/env_config_parser.h"
#include "mindspore/core/base/base.h"
#include "mindspore/core/ir/func_graph.h"

namespace mindspore {



+ 6
- 0
mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc View File

@@ -56,6 +56,9 @@
#include "tdt/tdt_host_interface.h"
#include "tdt/status.h"
#endif
#ifdef ENABLE_DUMP_IR
#include "debug/rdr/running_data_recorder.h"
#endif

using ge::model_runner::ModelRunner;
using mindspore::device::ascend::ProfilingManager;
@@ -480,6 +483,9 @@ bool AscendKernelRuntime::LoadTask(const session::KernelGraph *graph) {

status = ModelRunner::Instance().DistributeTask(model_iter->first);
if (!status) {
#ifdef ENABLE_DUMP_IR
mindspore::RDR::TriggerAll();
#endif
MS_LOG(EXCEPTION) << "Distribute Task Failed";
}



Loading…
Cancel
Save