Browse Source

fix(dnn): fix cudnn include

GitOrigin-RevId: f6f9731c3e
HuaHua404-patch-4
Megvii Engine Team 2 years ago
parent
commit
43bd949af0
2 changed files with 2 additions and 2 deletions
  1. +1
    -2
      dnn/src/cuda/cudnn_wrapper_v8.cpp
  2. +1
    -0
      dnn/src/cuda/cudnn_wrapper_v8.h

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

@@ -1,6 +1,5 @@
#if CUDNN_VERSION >= 8020

#include "src/cuda/cudnn_wrapper_v8.h" #include "src/cuda/cudnn_wrapper_v8.h"
#if CUDNN_VERSION >= 8020
#include "src/cuda/cudnn_wrapper.h" #include "src/cuda/cudnn_wrapper.h"


#include "src/common/utils.h" #include "src/common/utils.h"


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

@@ -1,4 +1,5 @@
#pragma once #pragma once
#include "src/cuda/cudnn_with_check.h"


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


Loading…
Cancel
Save