def current_stream_wait()

in deep_ep/utils.py [0:0]

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

    def current_stream_wait(self) -> None:
        """
        The current stream `torch.cuda.current_stream()` waits for the event to be finished.
        """
        assert self.event is not None
        self.event.current_stream_wait()