1#pragma once
2
3#include <ATen/cudnn/cudnn-wrapper.h>
4
5// Note: The version below should not actually be 8000. Instead, it should
6// be whatever version of cuDNN that v8 API work with PyTorch correctly.
7// The version is set to 8000 today for convenience of debugging.
8#if defined(USE_EXPERIMENTAL_CUDNN_V8_API) && defined(CUDNN_VERSION) && CUDNN_VERSION >= 8000
9#define HAS_CUDNN_V8() true
10#else
11#define HAS_CUDNN_V8() false
12#endif
13