#pragma once

#include <ATen/cudnn/cudnn-wrapper.h>
#include <ATen/Tensor.h>

namespace at { namespace native {

cudnnDataType_t getCudnnDataType(const at::Tensor& tensor);

int64_t cudnn_version();

}}

