Skip to content

Commit aa4e63d

Browse files
committed
refactor: use public priority aggregate fields
1 parent 7136945 commit aa4e63d

3 files changed

Lines changed: 8 additions & 8 deletions

File tree

cluster-trace-gpu-v2026/scripts/build_aggregates.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -428,8 +428,8 @@ def write_job_type_outputs(
428428
rows = [
429429
{
430430
"job_type": job_type,
431-
"used_card_hour_sum_p9300": values[0],
432-
"used_card_hour_sum_p9200_9300": values[1],
431+
"hp_used_gpu_hours": values[0],
432+
"lp_used_gpu_hours": values[1],
433433
}
434434
for job_type, values in sorted(job_type_priority.items())
435435
]
@@ -460,9 +460,9 @@ def write_gpu_hourly_req(gpu_hourly_req: dict[tuple[str, str], list[float]], out
460460
{
461461
"server_gpu_spec": spec,
462462
"day": day,
463-
"used_card_hour_sum_p9300": vals[0],
464-
"used_card_hour_sum_p9200_9300": vals[1],
465-
"used_card_hour_sum": vals[2],
463+
"hp_used_gpu_hours": vals[0],
464+
"lp_used_gpu_hours": vals[1],
465+
"total_used_gpu_hours": vals[2],
466466
}
467467
for (spec, day), vals in sorted(gpu_hourly_req.items())
468468
]

cluster-trace-gpu-v2026/scripts/plotters/gpu_spec_hourly_req_trend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def plot() -> None:
4343
for spec in plot_order:
4444
group = df[df["display_gpu_spec"].eq(spec)].sort_values("day")
4545
plot_x = [plot_x_by_day[day] for day in group["day"]]
46-
plot_y = group["used_card_hour_sum_p9300"].to_numpy(dtype=float) / 100.0
46+
plot_y = group["hp_used_gpu_hours"].to_numpy(dtype=float) / 100.0
4747
ax.plot(
4848
plot_x,
4949
plot_y,

cluster-trace-gpu-v2026/scripts/plotters/job_type_priority_distribution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@ def plot() -> None:
1717
rows = []
1818
for job_type in c.PAPER_JOB_ORDER:
1919
cur = df[df["job_type"].eq(job_type)]
20-
hp = float(cur["used_card_hour_sum_p9300"].sum())
21-
lp = float(cur["used_card_hour_sum_p9200_9300"].sum())
20+
hp = float(cur["hp_used_gpu_hours"].sum())
21+
lp = float(cur["lp_used_gpu_hours"].sum())
2222
total = hp + lp
2323
rows.append(
2424
{

0 commit comments

Comments
 (0)