Method current_stream

tensorrt_llm/python_plugin.py:319–323

(self)

    @property
    def current_stream(self):
        if self._current_stream == -1:
            return torch.cuda.current_stream().cuda_stream
        else:
            return self._current_stream

    @current_stream.setter
    def current_stream(self, stream: int):