(self, cuda_stream=None)
| 89 | |
| 90 | class TorchStream: |
| 91 | def __init__(self, cuda_stream=None): |
| 92 | if cuda_stream: |
| 93 | self.m_stream = torch.cuda.ExternalStream(cuda_stream) |
| 94 | else: |
| 95 | self.m_stream = torch.cuda.Stream() |
| 96 | |
| 97 | def cuda_stream(self): |
| 98 | return self.m_stream.cuda_stream |
nothing calls this directly
no test coverage detected