| 83 | } |
| 84 | |
| 85 | cudaError_t XRtEventQuery(cudaEvent_t event) |
| 86 | { |
| 87 | XDEBG("XRtEventQuery(event: %p)", event); |
| 88 | if (event == nullptr) return RtDriver::EventQuery(event); |
| 89 | auto xevent = g_events.Get(event, nullptr); |
| 90 | if (xevent == nullptr) return RtDriver::EventQuery(event); |
| 91 | |
| 92 | auto state = xevent->GetState(); |
| 93 | if (state >= kCommandStateCompleted) return cudaSuccess; |
| 94 | return cudaErrorNotReady; |
| 95 | } |
| 96 | |
| 97 | static cudaError_t SubmitEventRecord(cudaEvent_t event, cudaStream_t stream, bool ptsz) |
| 98 | { |