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

Pin the number of CPUs in failing actor test. (#2368)

Browse Source 
* Pin the number of CPUs in failing actor test.

* Pin number of CPUs in multi_node_test.py.

* Fix linting.
This commit is contained in:
Robert Nishihara
2018-07-11 20:34:19 -05:00
committed by Philipp Moritz
parent 4dadc60968
commit 54487b1d7f
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/actor_test.py
+1 -1
View File
@@ -484,7 +484,7 @@ class ActorNesting(unittest.TestCase):
def testRemoteFunctionWithinActor(self):
# Make sure we can use remote funtions within actors.
ray.init(num_cpus=100)
ray.init(num_cpus=10)
# Create some values to close over.
val1 = 1
test/multi_node_test.py
+6 -6
View File
@@ -223,7 +223,7 @@ class StartRayScriptTest(unittest.TestCase):
subprocess.Popen(["ray", "stop"]).wait()
# Test starting Ray with the number of CPUs specified.
run_and_get_output(["ray", "start", "--head", "--num-cpus", "100"])
run_and_get_output(["ray", "start", "--head", "--num-cpus", "2"])
subprocess.Popen(["ray", "stop"]).wait()
# Test starting Ray with the number of GPUs specified.
@@ -245,11 +245,11 @@ class StartRayScriptTest(unittest.TestCase):
# Test starting Ray with all arguments specified.
run_and_get_output([
"ray", "start", "--head", "--num-workers", "20",
"--redis-port", "6379", "--redis-shard-ports",
"6380,6381,6382", "--object-manager-port", "12345",
"--num-cpus", "100", "--num-gpus", "0", "--redis-max-clients",
"100", "--resources", "{\"Custom\": 1}"
"ray", "start", "--head", "--num-workers", "2", "--redis-port",
"6379", "--redis-shard-ports", "6380,6381,6382",
"--object-manager-port", "12345", "--num-cpus", "2",
"--num-gpus", "0", "--redis-max-clients", "100", "--resources",
"{\"Custom\": 1}"
])
subprocess.Popen(["ray", "stop"]).wait()