From 6ab1c55d2ca21ddb3b918842425b8fe8f6639011 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 12 May 2021 14:26:50 +0800
Subject: [PATCH] fix(mgb): fix fastrun workspace limit

GitOrigin-RevId: b9797eefcff8f970aebd0b6e5a2e44e25fe6b0d4
---
 src/opr/impl/search_policy/algo_chooser.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/opr/impl/search_policy/algo_chooser.cpp b/src/opr/impl/search_policy/algo_chooser.cpp
index 4ac6a7f7..4fd7a6b9 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<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;
         }
 
-- 
GitLab