From b98019977c5aaedf04dbb25c38e12572e668e777 Mon Sep 17 00:00:00 2001
From: Yaliang Wu <ylwu@amazon.com>
Date: Mon, 22 Jul 2024 12:44:37 -0700
Subject: [PATCH] increase execute thread pool size (#2691)

Signed-off-by: Yaliang Wu <ylwu@amazon.com>
---
 .../java/org/opensearch/ml/plugin/MachineLearningPlugin.java  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java b/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java
index 863c7cd054..8a8d1811d1 100644
--- a/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java
+++ b/plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java
@@ -842,8 +842,8 @@ public List<ExecutorBuilder<?>> getExecutorBuilders(Settings settings) {
         FixedExecutorBuilder executeThreadPool = new FixedExecutorBuilder(
             settings,
             EXECUTE_THREAD_POOL,
-            Math.max(1, OpenSearchExecutors.allocatedProcessors(settings) - 1),
-            10,
+            OpenSearchExecutors.allocatedProcessors(settings) * 4,
+            10000,
             ML_THREAD_POOL_PREFIX + EXECUTE_THREAD_POOL,
             false
         );