def capture()

in deep_ep/buffer.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

    def capture() -> EventOverlap:
        """
        Capture a CUDA event on the current stream, i.e. `torch.cuda.current_stream()`.

        Returns:
            event: the captured event.
        """
        return EventOverlap(EventHandle())