WaitForCudaStream block waits until all pending operations are finished on the specified Cuda stream.
(stream unsafe.Pointer, device int)
| 83 | // WaitForCudaStream block waits until all pending operations are finished on |
| 84 | // the specified Cuda stream. |
| 85 | func WaitForCudaStream(stream unsafe.Pointer, device int) { |
| 86 | if stream != nil { |
| 87 | doCGoCall(func() C.CGoCallResHandle { |
| 88 | return C.WaitForCudaStream(stream, C.int(device)) |
| 89 | }) |
| 90 | } |
| 91 | } |
| 92 | |
| 93 | // DestroyCudaStream destroys the specified Cuda stream. |
| 94 | func DestroyCudaStream(stream unsafe.Pointer, device int) { |
no test coverage detected