fix(mgb): fix fastrun workspace limit

GitOrigin-RevId: b9797eefcff8f970aebd0b6e5a2e44e25fe6b0d4

fix(mgb): fix fastrun workspace limit
GitOrigin-RevId: b9797eefcff8f970aebd0b6e5a2e44e25fe6b0d4
6ab1c55d · Megvii Engine Team · 7af1ec66 · 6ab1c55d
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

src/opr/impl/search_policy/algo_chooser.cpp src/opr/impl/search_policy/algo_chooser.cpp +1 -1

未找到文件。
--- a/src/opr/impl/search_policy/algo_chooser.cpp
+++ b/src/opr/impl/search_policy/algo_chooser.cpp
@@ -705,7 +705,7 @@ void AlgoChooser<Opr>::AlgoChooserHelper::profile(
        construct_execution_policy(selected_strategy, policy);
        mgb_assert(policy.algo.valid(),
                   "construct execution policy must success when profiling");
-        if (get_workspace_size_bytes(policy) >= workspace_limit) {
+        if (get_workspace_size_bytes(policy) > workspace_limit) {
            continue;
        }