Browse Source

fix(mgb): fix fastrun workspace limit

GitOrigin-RevId: b9797eefcf
release-1.4
Megvii Engine Team 4 years ago
parent
commit
0368eff688
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/opr/impl/search_policy/algo_chooser.cpp

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

@@ -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;
}



Loading…
Cancel
Save