hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Core] Fix Used Memory Calculation (#20127)

Browse source 
* fix memory

* fix
This commit is contained in:
Lixin Wei 2021-11-09 09:36:32 +08:00 • committed by GitHub
parent 9c2b8c8501
commit 8e666ca1e9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.gitignore vendored
View file

@ -214,3 +214,6 @@ test_state.json
# workflow storage # workflow storage
workflow_data/ workflow_data/
# vscode java extention generated
.factorypath

7
python/ray/_private/memory_monitor.py
View file

@ -127,12 +127,7 @@ class MemoryMonitor:
def get_memory_usage(self): def get_memory_usage(self):
psutil_mem = psutil.virtual_memory() psutil_mem = psutil.virtual_memory()
total_gb = psutil_mem.total / (1024**3) total_gb = psutil_mem.total / (1024**3)
used_gb = total_gb - psutil_mem.available / (1024**3) used_gb = psutil_mem.used / (1024**3)
# Linux, BSD has cached memory, which should
# also be considered as unused memory
if hasattr(psutil_mem, "cached"):
used_gb -= psutil_mem.cached / (1024**3)
if self.cgroup_memory_limit_gb < total_gb: if self.cgroup_memory_limit_gb < total_gb:
total_gb = self.cgroup_memory_limit_gb total_gb = self.cgroup_memory_limit_gb