Skip to content

Commit 4b8bc7a

Browse files
committed
increase warmup time
1 parent f4adf5f commit 4b8bc7a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/examples/test_load_latency.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,8 @@ def ping_pong(
6767
BLOCK_SIZE = 1
6868
BUFFER_LEN = 1
6969

70-
iter = 200
71-
skip = 1
70+
iter = 100
71+
skip = 10
7272
mm_begin_timestamp = torch.zeros((num_ranks, BLOCK_SIZE), dtype=torch.int64, device="cuda")
7373
mm_end_timestamp = torch.zeros((num_ranks, BLOCK_SIZE), dtype=torch.int64, device="cuda")
7474

0 commit comments

Comments
 (0)