Skip to content

Commit

Permalink
do not specify broker_is_worker when spawn mode is requested for dist…
Browse files Browse the repository at this point in the history
…ributed eval
  • Loading branch information
iraikov committed Aug 27, 2024
1 parent d574d3f commit ae23c34
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion distgfs.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import copy
import pprint
import importlib
import logging
import os
Expand Down Expand Up @@ -1069,10 +1070,11 @@ def run(
del gfsopt_params["file_path"]
if "save" in gfsopt_params:
del gfsopt_params["save"]

distwq.run(
fun_name="gfswork",
module_name="distgfs",
broker_is_worker=True,
broker_is_worker=not spawn_workers,
broker_fun_name=gfsopt_params.get("broker_fun_name", None),
broker_module_name=gfsopt_params.get("broker_module_name", None),
verbose=verbose,
Expand Down

0 comments on commit ae23c34

Please sign in to comment.