diff --git a/src/opr/impl/search_policy/algo_chooser.cpp b/src/opr/impl/search_policy/algo_chooser.cpp index 4ac6a7f7b914f495d0ecf59e7f28550cbf055396..4fd7a6b9f2f7fa633f701c33f9d32689f9b3d268 100644 --- a/src/opr/impl/search_policy/algo_chooser.cpp +++ b/src/opr/impl/search_policy/algo_chooser.cpp @@ -705,7 +705,7 @@ void AlgoChooser::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; }