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

[hotfix] CPU Detection (#10821)

Browse Source
This commit is contained in:
Alex Wu
2020-09-16 21:02:52 -07:00
committed by GitHub
parent 77c3414bc2
commit 6f479d4697
2 changed files with 107 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_advanced_3.py
+51
View File
@@ -4,6 +4,7 @@ import logging
import os
import sys
import socket
import tempfile
import time
import numpy as np
@@ -736,6 +737,56 @@ def test_accelerator_type_api(shutdown_only):
assert ray.available_resources()[resource_name] < quantity
def test_detect_docker_cpus():
# No limits set
with tempfile.NamedTemporaryFile(
"w") as quota_file, tempfile.NamedTemporaryFile(
"w") as period_file, tempfile.NamedTemporaryFile(
"w") as cpuset_file:
quota_file.write("-1")
period_file.write("100000")
cpuset_file.write("0-63")
quota_file.flush()
period_file.flush()
cpuset_file.flush()
assert ray.utils._get_docker_cpus(
cpu_quota_file_name=quota_file.name,
cpu_share_file_name=period_file.name,
cpuset_file_name=cpuset_file.name) == 64
# No cpuset used
with tempfile.NamedTemporaryFile(
"w") as quota_file, tempfile.NamedTemporaryFile(
"w") as period_file, tempfile.NamedTemporaryFile(
"w") as cpuset_file:
quota_file.write("-1")
period_file.write("100000")
cpuset_file.write("0-10,20,50-63")
quota_file.flush()
period_file.flush()
cpuset_file.flush()
assert ray.utils._get_docker_cpus(
cpu_quota_file_name=quota_file.name,
cpu_share_file_name=period_file.name,
cpuset_file_name=cpuset_file.name) == 26
# Quota set
with tempfile.NamedTemporaryFile(
"w") as quota_file, tempfile.NamedTemporaryFile(
"w") as period_file, tempfile.NamedTemporaryFile(
"w") as cpuset_file:
quota_file.write("42")
period_file.write("100")
cpuset_file.write("0-63")
quota_file.flush()
period_file.flush()
cpuset_file.flush()
assert ray.utils._get_docker_cpus(
cpu_quota_file_name=quota_file.name,
cpu_share_file_name=period_file.name,
cpuset_file_name=cpuset_file.name) == 0.42
if __name__ == "__main__":
import pytest
sys.exit(pytest.main(["-v", __file__]))