Show working set memory in dstack stats (#1856)

dstackai · Oct 17, 2024 · a171ba6 · a171ba6
1 parent 171ecc3
commit a171ba6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dstack/_internal/cli/commands/stats.py b/src/dstack/_internal/cli/commands/stats.py
@@ -97,7 +97,7 @@ def _get_stats_table(run: Run, metrics: List[JobMetrics]) -> Table:
         cpu_usage = _get_metric_value(job_metrics, "cpu_usage_percent")
         if cpu_usage is not None:
             cpu_usage = f"{cpu_usage}%"
-        memory_usage = _get_metric_value(job_metrics, "memory_usage_bytes")
+        memory_usage = _get_metric_value(job_metrics, "memory_working_set_bytes")
         if memory_usage is not None:
             memory_usage = f"{round(memory_usage / 1024 / 1024)}MB"
             if job.job_submissions[-1].job_provisioning_data is not None: