Merge pull request #31104 from vespa-engine/havardpe/upper-limit-on-total-size-of-compiled-gbdt-models

Avoid compiling gbdt models with many conditions

1 files changed, 3 insertions, 0 deletions

diff --git a/eval/src/vespa/eval/eval/gbdt.cpp b/eval/src/vespa/eval/eval/gbdt.cpp
index 3422228b03c..7ab4c4ae822 100644
--- a/eval/src/vespa/eval/eval/gbdt.cpp
+++ b/eval/src/vespa/eval/eval/gbdt.cpp
@@ -154,6 +154,9 @@ Optimize::select_best(const ForestStats &stats,
     if ((stats.tree_sizes.back().size > 12) && (path_len > 2500.0)) {
         return apply_chain(VMForest::optimize_chain, stats, trees);
     }
+    if (stats.total_size > 25000) {
+        return apply_chain(VMForest::optimize_chain, stats, trees);
+    }
     return Optimize::Result();
 }