| 329 | } |
| 330 | |
| 331 | Status CudaDevice::Stream::Synchronize() const { |
| 332 | ContextSaver set_temporary(reinterpret_cast<CUcontext>(context_.get()->handle())); |
| 333 | CU_RETURN_NOT_OK("cuStreamSynchronize", cuStreamSynchronize(value())); |
| 334 | return Status::OK(); |
| 335 | } |
| 336 | |
| 337 | Status CudaDevice::SyncEvent::Wait() { |
| 338 | ContextSaver set_temporary(reinterpret_cast<CUcontext>(context_.get()->handle())); |