mirror of
https://github.com/wassname/ray.git
synced 2026-09-12 12:51:15 +08:00
Fix stack overflow if many objects are fetched. (#237)
* fix stack overflow if many objects are fetched * fix other stack allocations * add tests and fix linting * address stephanie's comments * fix linting * fix tests
This commit is contained in:
committed by
Robert Nishihara
parent
e5a9fc0032
commit
ca254b8689
@@ -81,6 +81,18 @@ class TaskTests(unittest.TestCase):
|
||||
self.assertTrue(ray.services.all_processes_alive())
|
||||
ray.worker.cleanup()
|
||||
|
||||
def testGettingManyObjects(self):
|
||||
ray.init()
|
||||
|
||||
@ray.remote
|
||||
def f():
|
||||
return 1
|
||||
|
||||
n = 10 ** 4 # TODO(pcm): replace by 10 ** 5 once this is faster
|
||||
l = ray.get([f.remote() for _ in range(n)])
|
||||
self.assertEqual(l, n * [1])
|
||||
ray.worker.cleanup()
|
||||
|
||||
def testWait(self):
|
||||
for num_local_schedulers in [1, 4]:
|
||||
for num_workers_per_scheduler in [4]:
|
||||
|
||||
Reference in New Issue
Block a user