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