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 |