1 #pragma once 2 3 #include <cudnn.h> 4 5 #define STRINGIFY(x) #x 6 #define STRING(x) STRINGIFY(x) 7 8 #if CUDNN_MAJOR < 8 || (CUDNN_MAJOR == 8 && CUDNN_MINOR < 5) 9 #pragma message("CuDNN v" STRING( \ 10 CUDNN_MAJOR) " found, but need at least CuDNN v8. You can get the latest version of CuDNN from https://developer.nvidia.com/cudnn or disable CuDNN with USE_CUDNN=0") 11 #pragma message "We strongly encourage you to move to 8.5 and above." 12 #pragma message "This message is intended to annoy you enough to update." 13 #endif 14 15 #undef STRINGIFY 16 #undef STRING 17