Searched defs:CUDAGraph (Results 1 – 5 of 5) sorted by relevance
21 struct TORCH_CUDA_CPP_API CUDAGraph { struct43 cudaGraphExec_t graph_exec_ = nullptr;45 static std::atomic<int> pending_event_queries;51 bool has_graph_ = false;53 bool has_graph_exec_ = false;57 CaptureId_t capture_id_ = -1;82 // in a capture to run on the same device, but this is a limitation of CUDAGraph, argument85 int capture_dev_;
66 CUDAGraph::CUDAGraph() in CUDAGraph() function in at::cuda::CUDAGraph
7 CUDAGraph section in **Background**
46 class CUDAGraph(torch._C._CUDAGraph): class
594 existing_cuda_graph: Optional[torch.cuda.CUDAGraph],