diff --git a/dnn/src/cuda/cudnn_wrapper_v8.cpp b/dnn/src/cuda/cudnn_wrapper_v8.cpp index 671cf140..cc1809be 100644 --- a/dnn/src/cuda/cudnn_wrapper_v8.cpp +++ b/dnn/src/cuda/cudnn_wrapper_v8.cpp @@ -1,6 +1,5 @@ -#if CUDNN_VERSION >= 8020 - #include "src/cuda/cudnn_wrapper_v8.h" +#if CUDNN_VERSION >= 8020 #include "src/cuda/cudnn_wrapper.h" #include "src/common/utils.h" diff --git a/dnn/src/cuda/cudnn_wrapper_v8.h b/dnn/src/cuda/cudnn_wrapper_v8.h index b5de6da9..575f43d0 100644 --- a/dnn/src/cuda/cudnn_wrapper_v8.h +++ b/dnn/src/cuda/cudnn_wrapper_v8.h @@ -1,4 +1,5 @@ #pragma once +#include "src/cuda/cudnn_with_check.h" #if CUDNN_VERSION >= 8020 #include "megdnn/basic_types.h"