mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!24646 RDR exports collected data on dataset_iterator_kernel error 

Merge pull request !24646 from liangyongxiong/rdr
This commit is contained in:
i-robot 2021-10-11 12:07:47 +00:00 committed by Gitee
parent f81ca1f020 e04ce642a7
commit 8f33979fe5
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/backend/kernel_compiler/gpu/data/dataset_iterator_kernel.cc
View File

@ -27,6 +27,9 @@
#endif
#include "runtime/device/gpu/gpu_buffer_mgr.h"
#include "runtime/device/gpu/gpu_common.h"
#ifdef ENABLE_DUMP_IR
#include "debug/rdr/running_data_recorder.h"
#endif
namespace mindspore {
namespace kernel {
@ -113,6 +116,9 @@ bool DatasetIteratorKernel::ReadDevice(void **addr, size_t *len) {
MS_LOG(INFO) << "Waiting for data...(" << repeat << " / 10)";
continue;
} else {
#ifdef ENABLE_DUMP_IR
mindspore::RDR::TriggerAll();
#endif
MS_LOG(EXCEPTION) << "Get data timeout";
}
}