xref: /aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/Handle.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #pragma once
2 
3 #include <ATen/cuda/ATenCUDAGeneral.h>
4 #include <ATen/cudnn/cudnn-wrapper.h>
5 
6 namespace at::native {
7 
8 TORCH_CUDA_CPP_API cudnnHandle_t getCudnnHandle();
9 } // namespace at::native
10