Skip to content

Commit ddda4eb

Browse files
committed
[Test] Try to fix CI flaky rotary test
1 parent 737bad0 commit ddda4eb

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

tests/test_rotary.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,12 @@ def assert_one_cuda_kernel_no_memcpy(prof):
135135
"(CUPTI unavailable); cannot verify kernel-count / no-memcpy."
136136
)
137137
names = cuda_event_names(prof)
138+
if not names:
139+
# Probe confirmed CUPTI works in this process, but kineto intermittently
140+
# delivers zero CUDA events for an individual profile block under load
141+
# (observed in containerized CI). The "kernel removed entirely" regression
142+
# this would mask is still caught by the numerical asserts in these tests.
143+
pytest.skip("torch.profiler captured no CUDA events for this run (kineto flake)")
138144
kernels = [name for name in names if not name.startswith("Memcpy")]
139145
memcpys = [name for name in names if name.startswith("Memcpy")]
140146
assert len(kernels) == 1, names

0 commit comments

Comments
 (0)