@@ -53,15 +53,16 @@ def compile_pdl_overlap_kernels(device):
5353
5454
5555def run_pdl_overlap_check (device , * , via_graph : bool = False ):
56- """Run the shared primary/secondary PDL overlap protocol.
56+ """Run the shared same-stream primary/secondary PDL overlap protocol.
5757
58- Asserts no overlap without ``programmatic_stream_serialization``, then retries
59- a few times with it enabled. Overlap is opportunistic → miss is xfail.
58+ Both paths launch primary then secondary on one stream. Asserts no overlap
59+ without ``programmatic_stream_serialization``, then retries a few times with
60+ it enabled. Overlap is opportunistic → miss is xfail.
6061
6162 Args:
6263 device: Current CUDA device (compute capability >= 9.0 required).
63- via_graph: If True, capture launches into a CUDA graph and launch the
64- graph; otherwise launch kernels directly on the stream .
64+ via_graph: If True, stream- capture the same-stream launches into a CUDA
65+ graph and launch that graph ; otherwise launch kernels directly.
6566 """
6667 if device .compute_capability < (9 , 0 ):
6768 pytest .skip ("Programmatic Dependent Launch requires compute capability >= 9.0" )
@@ -97,7 +98,7 @@ def _run(secondary_launch_cfg: LaunchConfig) -> int:
9798 stream .sync ()
9899 return int (overlapped [0 ])
99100
100- path = "graph-capture " if via_graph else "same-stream"
101+ path = "same-stream (graph) " if via_graph else "same-stream"
101102 assert _run (secondary_serial_cfg ) == 0 , (
102103 f"Expected no overlap when programmatic_stream_serialization is False ({ path } )"
103104 )
0 commit comments