From c159d7670ba3c817c449bc1e0500a4645967c93f Mon Sep 17 00:00:00 2001 From: Paul Date: Thu, 19 Sep 2024 10:24:12 -0500 Subject: [PATCH 1/2] Add flag to disable pass --- src/pass_manager.cpp | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/pass_manager.cpp b/src/pass_manager.cpp index af8f0e4d6a7..f2e58a03035 100644 --- a/src/pass_manager.cpp +++ b/src/pass_manager.cpp @@ -40,6 +40,14 @@ inline namespace MIGRAPHX_INLINE_NS { MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_TRACE_PASSES); MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_TIME_PASSES); +MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_DISABLE_PASSES); + +static bool is_pass_disabled(const std::string& name) +{ + static const auto passes = + split_string(string_value_of(MIGRAPHX_DISABLE_PASSES{}, ""), ','); + return contains(passes, name); +} void validate_pass(module& mod, const pass& p, tracer trace) { @@ -126,6 +134,8 @@ struct module_pm : module_pass_manager virtual void run_pass(const pass& p) override { + if(is_pass_disabled(p.name())) + return; trace("Pass: ", p.name()); assert(mod); assert(mod->validate() == mod->end()); From 982023cadf624a222a3220f0b50844e6457aae24 Mon Sep 17 00:00:00 2001 From: Paul Date: Thu, 19 Sep 2024 10:27:40 -0500 Subject: [PATCH 2/2] Format --- src/pass_manager.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/pass_manager.cpp b/src/pass_manager.cpp index f2e58a03035..f950277bd0a 100644 --- a/src/pass_manager.cpp +++ b/src/pass_manager.cpp @@ -44,8 +44,7 @@ MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_DISABLE_PASSES); static bool is_pass_disabled(const std::string& name) { - static const auto passes = - split_string(string_value_of(MIGRAPHX_DISABLE_PASSES{}, ""), ','); + static const auto passes = split_string(string_value_of(MIGRAPHX_DISABLE_PASSES{}, ""), ','); return contains(passes, name); }