mirror of
https://github.com/wassname/ray.git
synced 2026-06-30 10:15:04 +08:00
Loadbalancing Test issue (#452)
* Limiting number of CPUs in loadbalancing test * fixes as requested
This commit is contained in:
committed by
Robert Nishihara
parent
fb4525f833
commit
c3a2505ffd
+4
-3
@@ -1328,10 +1328,11 @@ class SchedulingAlgorithm(unittest.TestCase):
|
||||
def testLoadBalancing(self):
|
||||
# This test ensures that tasks are being assigned to all local schedulers
|
||||
# in a roughly equal manner.
|
||||
num_workers = 21
|
||||
num_local_schedulers = 3
|
||||
ray.worker._init(start_ray_local=True, num_workers=num_workers,
|
||||
num_local_schedulers=num_local_schedulers)
|
||||
num_cpus = 7
|
||||
ray.worker._init(start_ray_local=True,
|
||||
num_local_schedulers=num_local_schedulers,
|
||||
num_cpus=num_cpus)
|
||||
|
||||
@ray.remote
|
||||
def f():
|
||||
|
||||
Reference in New Issue
Block a user