@@ -55,7 +55,9 @@ def __init__(
55
55
user_endpoint_config:
56
56
User endpoint configuration values as described
57
57
and allowed by endpoint administrators. Must be a JSON-serializable dict
58
- or None.
58
+ or None. Refer docs from `globus-compute
59
+ <https://globus-compute.readthedocs.io/en/latest/endpoints/endpoints.html#templating-endpoint-configuration>`_
60
+ for more info.
59
61
60
62
label:
61
63
a label to name the executor; mainly utilized for
@@ -119,11 +121,16 @@ def submit(self, func: Callable, resource_specification: Dict[str, Any], *args:
119
121
120
122
func: Callable
121
123
Python function to execute remotely
124
+
122
125
resource_specification: Dict[str, Any]
123
- Resource specification used to run MPI applications on Endpoints configured
124
- to use globus compute's MPIEngine
126
+ Resource specification can be used specify MPI resources required by MPI applications on
127
+ Endpoints configured to use globus compute's MPIEngine. GCE also accepts `user_endpoint_config`
128
+ to configure endpoints when the endpoint is a `Multi-User Endpoint
129
+ <https://globus-compute.readthedocs.io/en/latest/endpoints/endpoints.html#templating-endpoint-configuration>`_
130
+
125
131
args:
126
132
Args to pass to the function
133
+
127
134
kwargs:
128
135
kwargs to pass to the function
129
136
@@ -133,6 +140,8 @@ def submit(self, func: Callable, resource_specification: Dict[str, Any], *args:
133
140
Future
134
141
"""
135
142
self ._executor .resource_specification = resource_specification or self .resource_specification
143
+ # Pop user_endpoint_config since it is illegal in resource_spec for globus_compute
144
+ self ._executor .user_endpoint_config = resource_specification .pop ('user_endpoint_config' , self .user_endpoint_config )
136
145
return self ._executor .submit (func , * args , ** kwargs )
137
146
138
147
def shutdown (self , wait = True , * , cancel_futures = False ):
0 commit comments