9dd3461
1
2
3
4
5
6
7
8
9
10
#pragma once #include <ATen/cudnn/cudnn-wrapper.h> #include <ATen/cuda/ATenCUDAGeneral.h> namespace at { namespace native { TORCH_CUDA_CPP_API cudnnHandle_t getCudnnHandle(); }} // namespace at::native
