Skip to content

Commit

Permalink
setting number of thread per worker to 8
Browse files Browse the repository at this point in the history
  • Loading branch information
gtramonte committed Jul 26, 2024
1 parent 7cda4fd commit 533fdff
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions executor/app/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -249,8 +249,9 @@ def create_dask_cluster(self, dask_cluster_opts: dict = None):
dask_cluster_opts["processes"] = True
port = int(os.getenv("DASK_DASHBOARD_PORT", 8787))
dask_cluster_opts["dashboard_address"] = f":{port}"
dask_cluster_opts["n_workers"] = 4
dask_cluster_opts["n_workers"] = 1
dask_cluster_opts["memory_limit"] = "auto"
dask_cluster_opts['thread_per_worker'] = 8
self._worker_id = self._db.create_worker(
status="enabled",
dask_scheduler_port=dask_cluster_opts["scheduler_port"],
Expand All @@ -266,7 +267,7 @@ def create_dask_cluster(self, dask_cluster_opts: dict = None):
#scheduler_port=dask_cluster_opts["scheduler_port"],
#dashboard_address=dask_cluster_opts["dashboard_address"],
#memory_limit=dask_cluster_opts["memory_limit"],
threads_per_worker=1
threads_per_worker=dask_cluster_opts['thread_per_worker'],
)
self._LOG.info(
"not creating Dask Client...", extra={"track_id": self._worker_id}
Expand Down

0 comments on commit 533fdff

Please sign in to comment.