(self, stream=None)
| 65 | raise RuntimeError(f"CUDA Error {str(err)}") |
| 66 | |
| 67 | def stop(self, stream=None): |
| 68 | if not stream: |
| 69 | stream = cuda.CUstream(0) |
| 70 | |
| 71 | (err,) = cuda.cuEventRecord(self.events[1], stream) |
| 72 | if err != cuda.CUresult.CUDA_SUCCESS: |
| 73 | raise RuntimeError(f"CUDA Error {str(err)}") |
| 74 | pass |
| 75 | |
| 76 | def stop_and_wait(self, stream=None): |
| 77 | if not stream: |