(self, stream=None)
| 57 | ] |
| 58 | |
| 59 | def start(self, stream=None): |
| 60 | if not stream: |
| 61 | stream = cuda.CUstream(0) |
| 62 | |
| 63 | (err,) = cuda.cuEventRecord(self.events[0], stream) |
| 64 | if err != cuda.CUresult.CUDA_SUCCESS: |
| 65 | raise RuntimeError(f"CUDA Error {str(err)}") |
| 66 | |
| 67 | def stop(self, stream=None): |
| 68 | if not stream: |