From 674f235a5bc4f4fc50c438e5b268431405745a1c Mon Sep 17 00:00:00 2001 From: w1049 Date: Tue, 1 Jul 2025 17:28:39 +0800 Subject: [PATCH] [Fix] Replace dmlc::Error with std::exception in VerifyGPUCode --- src/meta_schedule/postproc/verify_gpu_code.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)) {