facebookresearch / facebookresearch/clusterscope

🚀[Feature Request]: GPU and CPU memory both should be queryable via ClusterScope

Open
#80 0 comments 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
24
Forks
6
PR merge metrics
No merged PRs in 30d

Description

### A clear and concise description of what you want to happen.

This method implicitly assumes that clients are only interested in GPU-memory. Let's add support for both CPU and GPU memory and rename existing method(s).

```
def get_mem_per_node_MB(self) -> int:
"""Return the lowest amount of mem configured across all nodes in the cluster. Returns 0 if not a Slurm cluster.
```

### A clear and concise description of any alternative solutions or features you've considered, if any.

_No response_

### Additional context

_No response_

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.