Updating exception handling to catch dmlc::error.

apache · Mar 26, 2019 · 570398d · 570398d
1 parent c04bf12
commit 570398d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cpp-package/example/utils.h b/cpp-package/example/utils.h
@@ -30,10 +30,13 @@ using namespace mxnet::cpp;
 #define TRY \
   try {
 #define CATCH \
-  } catch(...) { \
+  } catch(dmlc::Error &err) { \
     LG << "Status: FAIL";\
     LG << "With Error: " << MXGetLastError(); \
     return 1; \
+  } catch(...) { \
+    LG << "Status: FAIL";\
+    return 1; \
   }
 
 bool isFileExists(const std::string &filename) {