diff --git a/src/meta_schedule/postproc/verify_gpu_code.cc b/src/meta_schedule/postproc/verify_gpu_code.cc index 8ffc424e4451..5b7b637ef242 100644 --- a/src/meta_schedule/postproc/verify_gpu_code.cc +++ b/src/meta_schedule/postproc/verify_gpu_code.cc @@ -31,7 +31,7 @@ class ThreadExtentChecker : private StmtVisitor { ThreadExtentChecker checker(thread_warp_size); checker.VisitStmt(stmt); return true; - } catch (const dmlc::Error& e) { + } catch (const std::exception&) { return false; } } @@ -188,7 +188,7 @@ class VerifyGPUCodeNode : public PostprocNode { } IRModule mod = IRModule(Map({{GlobalVar(g_var->name_hint), f}})); lowered = tvm::transform::Sequential(pass_list)(std::move(mod)); - } catch (const dmlc::Error& e) { + } catch (const std::exception&) { return false; } if (!Verify(lowered)) {