torch.profiler_allow_cudagraph_cupti_lazy_reinit_cuda12
function profiler_allow_cudagraph_cupti_lazy_reinit_cuda12(): voidAllows CUDA graph lazy reinitialization for NVIDIA CUDA 12.
CUDA graphs are a feature for optimizing GPU operations by capturing and replaying sequences of operations. This function controls lazy reinitialization behavior for NVIDIA's CUPTI profiler integration with CUDA 12.
This is a no-op in torch.js. CUDA and CUPTI profiling are not available in WebGPU. Use browser profiling tools or WebGPU profiling extensions instead.
Returns
void
Examples
// Enable lazy reinitialization for CUDA graph profiling
torch.profiler_allow_cudagraph_cupti_lazy_reinit_cuda12();See Also
- PyTorch torch.profiler.profiler_allow_cudagraph_cupti_lazy_reinit_cuda12()
- profiler - PyTorch profiler module