wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 08:07:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Speed up task submissions a bit (#5992)

Browse Source
This commit is contained in:
Eric Liang
2019-10-25 00:10:37 -07:00
committed by GitHub
parent 6f27d881bd
commit 4edae7ea2b
4 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/remote_function.py
+4 -2
View File
@@ -57,6 +57,8 @@ class RemoteFunction(object):
object_store_memory, resources, num_return_vals, max_calls):
self._function = function
self._function_descriptor = FunctionDescriptor.from_function(function)
self._function_descriptor_list = (
self._function_descriptor.get_function_descriptor_list())
self._function_name = (
self._function.__module__ + "." + self._function.__name__)
self._num_cpus = (DEFAULT_REMOTE_FUNCTION_CPUS
@@ -149,8 +151,8 @@ class RemoteFunction(object):
num_return_vals)
else:
object_ids = worker.core_worker.submit_task(
self._function_descriptor.get_function_descriptor_list(),
list_args, num_return_vals, resources)
self._function_descriptor_list, list_args, num_return_vals,
resources)
if len(object_ids) == 1:
return object_ids[0]