Searched refs:currentStreamCaptureStatusMayInitCtx (Results 1 – 5 of 5) sorted by relevance
24 return c10::cuda::currentStreamCaptureStatusMayInitCtx(); in currentStreamCaptureStatus()
70 inline CaptureStatus currentStreamCaptureStatusMayInitCtx() { in currentStreamCaptureStatusMayInitCtx() function
875 if (at::cuda::currentStreamCaptureStatusMayInitCtx() == in cudaMallocMaybeCapturing()
707 const bool is_capturing = at::cuda::currentStreamCaptureStatusMayInitCtx() != in split_with_sizes_copy_out_cuda()
2612 c10::cuda::currentStreamCaptureStatusMayInitCtx(); in checkForNCCLErrorsInternal()2679 c10::cuda::currentStreamCaptureStatusMayInitCtx(); in checkForNCCLErrorsInternal()2851 c10::cuda::currentStreamCaptureStatusMayInitCtx(); in checkForNCCLErrorsInternal()3255 c10::cuda::currentStreamCaptureStatusMayInitCtx(); in checkForNCCLErrorsInternal()