From 1dc50a2302db1cc6a04ff6010fb835a9d7e4a3dd Mon Sep 17 00:00:00 2001
From: xiefangqi <xiefangqi2@huawei.com>
Date: Mon, 28 Sep 2020 10:39:06 +0800
Subject: [PATCH] minddata catch jpeg errexit runtime error

---
 .../ccsrc/minddata/dataset/kernels/image/image_utils.cc   | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc b/mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc
index 2907fbb829a..80947349157 100644
--- a/mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc
+++ b/mindspore/ccsrc/minddata/dataset/kernels/image/image_utils.cc
@@ -161,7 +161,13 @@ static void JpegInitSource(j_decompress_ptr cinfo) {}
 
 static boolean JpegFillInputBuffer(j_decompress_ptr cinfo) {
   if (cinfo->src->bytes_in_buffer == 0) {
-    ERREXIT(cinfo, JERR_INPUT_EMPTY);
+    // Under ARM platform raise runtime_error may cause core problem,
+    // so we catch runtime_error and just return FALSE.
+    try {
+      ERREXIT(cinfo, JERR_INPUT_EMPTY);
+    } catch (std::runtime_error &e) {
+      return FALSE;
+    }
     return FALSE;
   }
   return TRUE;