From e04ce642a7b24dbfc03c18fca55f52c1dff3332f Mon Sep 17 00:00:00 2001
From: liangyongxiong <liangyongxiong1@huawei.com>
Date: Mon, 11 Oct 2021 11:07:30 +0800
Subject: [PATCH] RDR exports collected data on dataset_iterator_kernel error

---
 .../kernel_compiler/gpu/data/dataset_iterator_kernel.cc     | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/data/dataset_iterator_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/gpu/data/dataset_iterator_kernel.cc
index 5b5ec7fab0c..5be09542d2c 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/gpu/data/dataset_iterator_kernel.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/data/dataset_iterator_kernel.cc
@@ -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";
       }
     }