| 335 | } |
| 336 | |
| 337 | Status CudaDevice::SyncEvent::Wait() { |
| 338 | ContextSaver set_temporary(reinterpret_cast<CUcontext>(context_.get()->handle())); |
| 339 | CU_RETURN_NOT_OK("cuEventSynchronize", cuEventSynchronize(value())); |
| 340 | return Status::OK(); |
| 341 | } |
| 342 | |
| 343 | Status CudaDevice::SyncEvent::Record(const Device::Stream& st) { |
| 344 | auto cuda_stream = checked_cast<const CudaDevice::Stream*, const Device::Stream*>(&st); |