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

[rllib/tune] Add test for fractional gpu support in xray mode; add rllib support for fractional gpu (#2768)

Browse Source 
* frac gpu

* doc

* Update rllib-training.rst

* yapf

* remove xray
This commit is contained in:
Eric Liang
2018-09-03 11:12:23 -07:00
committed by GitHub
parent 9d655721e5
commit df4788e501
10 changed files with 41 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/test/trial_runner_test.py
+23
View File
@@ -800,6 +800,29 @@ class TrialRunnerTest(unittest.TestCase):
self.assertEqual(trials[0].status, Trial.TERMINATED)
self.assertEqual(trials[1].status, Trial.PENDING)
def testFractionalGpus(self):
ray.init(num_cpus=4, num_gpus=1, use_raylet=True)
runner = TrialRunner(BasicVariantGenerator())
kwargs = {
"resources": Resources(cpu=1, gpu=0.5),
}
trials = [
Trial("__fake", **kwargs),
Trial("__fake", **kwargs),
Trial("__fake", **kwargs),
Trial("__fake", **kwargs)
]
for t in trials:
runner.add_trial(t)
for _ in range(10):
runner.step()
self.assertEqual(trials[0].status, Trial.RUNNING)
self.assertEqual(trials[1].status, Trial.RUNNING)
self.assertEqual(trials[2].status, Trial.PENDING)
self.assertEqual(trials[3].status, Trial.PENDING)
def testResourceScheduler(self):
ray.init(num_cpus=4, num_gpus=1)
runner = TrialRunner(BasicVariantGenerator())