| 44 | } |
| 45 | |
| 46 | void EnqueueBlockingCallback(const c10::cuda::CUDAStream& stream, |
| 47 | StreamCallbackGate* gate) { |
| 48 | C10_CUDA_CHECK(hipStreamAddCallback( |
| 49 | stream.raw_stream(), BlockingStreamCallback, gate, 0)); |
| 50 | } |
| 51 | |
| 52 | void CreateRawStream(hipStream_t* stream) { |
| 53 | C10_CUDA_CHECK(hipStreamCreate(stream)); |