Skip to content

Commit 09668ba

Browse files
Apply Ruff auto-fixes
1 parent 46fc083 commit 09668ba

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

benchmarks/bench_ping_pong_latency.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def ping_pong(
4040
start = read_realtime()
4141
tl.store(mm_begin_timestamp_ptr + peer_rank * BLOCK_SIZE + offsets, start, time_stmp_mask)
4242
first_rank = tl.minimum(curr_rank, peer_rank) if (i % 2) == 0 else tl.maximum(curr_rank, peer_rank)
43-
token_first_done = i + 1
43+
token_first_done = i + 1
4444
token_second_done = i + 2
4545
if curr_rank == first_rank:
4646
iris.store(data + offsets, token_first_done, curr_rank, peer_rank, heap_bases, mask=data_mask)
@@ -144,6 +144,7 @@ def _pretty_print_matrix(latency_matrix: torch.Tensor) -> None:
144144

145145
def _write_csv(path: Path, latency_matrix: torch.Tensor) -> None:
146146
import csv
147+
147148
path.parent.mkdir(parents=True, exist_ok=True)
148149
with path.open("w", newline="") as f:
149150
writer = csv.writer(f)
@@ -186,7 +187,6 @@ def save_results(latency_matrix: torch.Tensor, out: str | None) -> None:
186187
raise ValueError(f"Unsupported output file extension: {out}")
187188

188189

189-
190190
def print_run_settings(
191191
args: dict,
192192
num_ranks: int,

0 commit comments

Comments
 (0)