| 13 | } |
| 14 | |
| 15 | TStreamCapture TStreamCapture::Capture(TCudaStream stream, |
| 16 | TCudaGraph* graph) { |
| 17 | CUDA_SAFE_CALL(cudaStreamBeginCapture(stream, cudaStreamCaptureModeThreadLocal)); |
| 18 | return TStreamCapture(graph, stream); |
| 19 | } |
| 20 | } |
no test coverage detected