Use dmlc error instead (apache#29)

jroesch · Aug 16, 2018 · 4481e48 · 4481e48
1 parent 2d8eeda
commit 4481e48
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 20 deletions.
diff --git a/relay/src/relay/evaluator.cc b/relay/src/relay/evaluator.cc
@@ -12,11 +12,8 @@ namespace relay {
 
 using namespace tvm::runtime;
 
-struct EvalError : std::exception {
-  std::string msg;
-  explicit EvalError(std::string msg) : msg(msg) {}
-
-  const char* what() const noexcept { return msg.c_str(); }
+struct EvalError : dmlc::Error {
+  explicit EvalError(const std::string & msg) : Error(msg) {}
 };
 
 Evaluator::Evaluator() : env() {}

diff --git a/relay/src/relay/forward_ad.cc b/relay/src/relay/forward_ad.cc
@@ -12,11 +12,8 @@ namespace relay {
 
 using namespace tvm::runtime;
 
-struct ForwardADError : std::exception {
-  std::string msg;
-  explicit ForwardADError(const std::string & msg) : msg(msg) {}
-
-  const char* what() const noexcept { return msg.c_str(); }
+struct ForwardADError : dmlc::Error {
+  explicit ForwardADError(const std::string & msg) : Error(msg) {}
 };
 
 // todo(M.K.) use let to share intermediate code

diff --git a/relay/src/relay/pretty_printer.cc b/relay/src/relay/pretty_printer.cc
@@ -14,11 +14,8 @@ namespace relay {
 
 using namespace tvm::runtime;
 
-struct PrintError : std::exception {
-  std::string msg;
-  explicit PrintError(std::string msg) : msg(msg) {}
-
-  const char* what() const noexcept { return msg.c_str(); }
+struct PrintError : dmlc::Error {
+  explicit PrintError(const std::string & msg) : Error(msg) {}
 };
 
 void PrettyPrinter::PrettyPrint(const Expr & expr, ostream & os) {

diff --git a/relay/src/relay/reverse_ad.cc b/relay/src/relay/reverse_ad.cc
@@ -12,11 +12,8 @@ namespace relay {
 
 using namespace tvm::runtime;
 
-struct ReverseADError : std::exception {
-  std::string msg;
-  explicit ReverseADError(const std::string & msg) : msg(msg) {}
-
-  const char* what() const noexcept { return msg.c_str(); }
+struct ReverseADError : dmlc::Error {
+  explicit ReverseADError(const std::string & msg) : Error(msg) {}
 };
 
 Expr ReverseAD::AD(const Expr& expr) { return this->operator()(expr); }