From 43bd949af05e919b6465f93129ff4eba4b522bc7 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 1 Aug 2022 16:43:07 +0800 Subject: [PATCH] fix(dnn): fix cudnn include GitOrigin-RevId: f6f9731c3e4949c80b26b570785e44432f0d63b7 --- dnn/src/cuda/cudnn_wrapper_v8.cpp | 3 +-- dnn/src/cuda/cudnn_wrapper_v8.h | 1 + 2 files changed, 2 insertions(+), 2 deletions(-) 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"