fix(dnn): fix cudnn include

GitOrigin-RevId: f6f9731c3e4949c80b26b570785e44432f0d63b7

fix(dnn): fix cudnn include
GitOrigin-RevId: f6f9731c3e4949c80b26b570785e44432f0d63b7
43bd949a · Megvii Engine Team · 8abc3ab8 · 43bd949a · 43bd949a
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

dnn/src/cuda/cudnn_wrapper_v8.cpp dnn/src/cuda/cudnn_wrapper_v8.cpp +1 -2

dnn/src/cuda/cudnn_wrapper_v8.h dnn/src/cuda/cudnn_wrapper_v8.h +1 -0

未找到文件。
--- a/dnn/src/cuda/cudnn_wrapper_v8.cpp
+++ b/dnn/src/cuda/cudnn_wrapper_v8.cpp
-#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"

--- a/dnn/src/cuda/cudnn_wrapper_v8.h
+++ b/dnn/src/cuda/cudnn_wrapper_v8.h
 #pragma once
+#include "src/cuda/cudnn_with_check.h"

 #if CUDNN_VERSION >= 8020
 #include "megdnn/basic_types.h"