From 63a6b0e710fded230f340a237e55085fb4ace4e9 Mon Sep 17 00:00:00 2001 From: Robert Nishihara Date: Tue, 30 Jul 2019 22:28:35 -0700 Subject: [PATCH] Fix bug in passing large arguments to tasks. (#5325) --- python/ray/_raylet.pyx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/ray/_raylet.pyx b/python/ray/_raylet.pyx index 67216f53b..abb216936 100644 --- a/python/ray/_raylet.pyx +++ b/python/ray/_raylet.pyx @@ -96,7 +96,7 @@ def compute_task_id(ObjectID object_id): return TaskID(object_id.native().TaskId().Binary()) -cdef c_bool is_simple_value(value, int *num_elements_contained): +cdef c_bool is_simple_value(value, int64_t *num_elements_contained): num_elements_contained[0] += 1 if num_elements_contained[0] >= RayConfig.instance().num_elements_limit(): @@ -170,7 +170,7 @@ def check_simple_value(value): True if the value should be send by value, False otherwise. """ - cdef int num_elements_contained = 0 + cdef int64_t num_elements_contained = 0 return is_simple_value(value, &num_elements_contained)