Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!7086 Only dump the specified kernels in dump config file, GPU mode 

Merge pull request !7086 from AdelShafiei/kernels_dump
This commit is contained in:
mindspore-ci-bot 2020-10-10 04:38:17 +08:00 committed by Gitee
parent 8bd60d1e3c aa33ae12fb
commit 14d1c4390b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/runtime/device/gpu/gpu_kernel_runtime.cc
View File

@ -88,12 +88,16 @@ void LoadKernelData(Debugger *debugger, const CNodePtr &kernel,
if (debugger) { if (debugger) {
debugger->SetCurNode(kernel_name); debugger->SetCurNode(kernel_name);
if (dump_enabled) { if (dump_enabled) {
read_data = true; auto &dump_json_parser = DumpJsonParser::GetInstance();
auto dump_mode = dump_json_parser.dump_mode();
// dump the node if dump_mode is 0, which means all kernels, or if this kernel is in the kernels list
if ((dump_mode == 0) || ((dump_mode == 1) && dump_json_parser.NeedDump(kernel_name))) {
read_data = true;
}
} else if (debugger->debugger_enabled()) { } else if (debugger->debugger_enabled()) {
read_data = debugger->ReadNodeDataRequired(); read_data = debugger->ReadNodeDataRequired();
} }
} }
if (!read_data) { if (!read_data) {
return; return;
} }