Searched refs:currentStreamCaptureStatus (Results 1 – 8 of 8) sorted by relevance
21 inline CaptureStatus currentStreamCaptureStatus() { in currentStreamCaptureStatus() function31 auto status = currentStreamCaptureStatus(); in assertNotCapturing()41 auto status = currentStreamCaptureStatus(); in errorIfCapturingCudnnBenchmark()
100 if (at::cuda::currentStreamCaptureStatus() != at::cuda::CaptureStatus::None) { in increase()450 if (at::cuda::currentStreamCaptureStatus() != at::cuda::CaptureStatus::None) { in philox_cuda_state()
501 if (at::cuda::currentStreamCaptureStatus() == at::cuda::CaptureStatus::None) { in mha_fwd()513 if (at::cuda::currentStreamCaptureStatus() != at::cuda::CaptureStatus::None) { in mha_fwd()754 if (at::cuda::currentStreamCaptureStatus() == at::cuda::CaptureStatus::None) { in mha_varlen_fwd()766 if (at::cuda::currentStreamCaptureStatus() != at::cuda::CaptureStatus::None) { in mha_varlen_fwd()987 if (at::cuda::currentStreamCaptureStatus() == in mha_bwd()1221 if (at::cuda::currentStreamCaptureStatus() == in mha_varlen_bwd()
178 if (at::cuda::currentStreamCaptureStatus() == at::cuda::CaptureStatus::None) {189 if (at::cuda::currentStreamCaptureStatus() != at::cuda::CaptureStatus::None) {200 if (at::cuda::currentStreamCaptureStatus() ==413 if (at::cuda::currentStreamCaptureStatus() ==
135 if (at::cuda::currentStreamCaptureStatus() == at::cuda::CaptureStatus::None) { in _use_cudnn_ctc_loss_tensor()
397 if (at::cuda::currentStreamCaptureStatus() == in _efficient_attention_backward()
1044 at::cuda::currentStreamCaptureStatus() != at::cuda::CaptureStatus::None; in _efficient_attention_forward()
1746 if (at::cuda::currentStreamCaptureStatus() == at::cuda::CaptureStatus::None) { in THCPModule_isCurrentStreamCapturing_wrap()