OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

fix(dnn): fix cudnn include 

GitOrigin-RevId: f6f9731c3e
HuaHua404-patch-4
Megvii Engine Team 2 years ago
parent
8abc3ab8fc
commit
43bd949af0
2 changed files with 2 additions and 2 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  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"
#if CUDNN_VERSION >= 8020
#include "src/cuda/cudnn_wrapper.h"

#include "src/common/utils.h"


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

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

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


Write Preview
Loading…
Cancel
Save