wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 15:56:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Ray, Tune, and RLlib support for memory, object_store_memory options (#5226)

Browse Source
This commit is contained in:
Eric Liang
2019-08-22 14:01:10 +08:00
committed by Robert Nishihara
parent c852213b83
commit e2e30ca507
40 changed files with 1006 additions and 296 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_multi_node_2.py
+9
View File
@@ -73,6 +73,15 @@ def verify_load_metrics(monitor, expected_resource_usage=None, timeout=10):
monitor.process_messages()
resource_usage = monitor.load_metrics.get_resource_usage()
if "memory" in resource_usage[1]:
del resource_usage[1]["memory"]
if "object_store_memory" in resource_usage[2]:
del resource_usage[1]["object_store_memory"]
if "memory" in resource_usage[2]:
del resource_usage[2]["memory"]
if "object_store_memory" in resource_usage[2]:
del resource_usage[2]["object_store_memory"]
if expected_resource_usage is None:
if all(x for x in resource_usage[1:]):
break