Browse Source

fix(mgb/cuda-stub): add libcuda-wrap_11.4.h to fit the CUDA11.4 toolchain

GitOrigin-RevId: efa38f00d1
tags/v1.8.0
Megvii Engine Team 3 years ago
parent
commit
3251f50114
2 changed files with 7840 additions and 0 deletions
  1. +7838
    -0
      dnn/cuda-stub/src/libcuda-wrap_11.4.h
  2. +2
    -0
      dnn/cuda-stub/src/libcuda.cpp

+ 7838
- 0
dnn/cuda-stub/src/libcuda-wrap_11.4.h
File diff suppressed because it is too large
View File


+ 2
- 0
dnn/cuda-stub/src/libcuda.cpp View File

@@ -56,6 +56,8 @@ CUresult on_init_failed(int func_idx) {
#include "./libcuda-wrap_11.1.h" #include "./libcuda-wrap_11.1.h"
#elif CUDA_VERSION == 11020 #elif CUDA_VERSION == 11020
#include "./libcuda-wrap_11.2.h" #include "./libcuda-wrap_11.2.h"
#elif CUDA_VERSION == 11040
#include "./libcuda-wrap_11.4.h"
#else #else
#error "cuda stub not support this cuda version, you can close cuda stub to passby" #error "cuda stub not support this cuda version, you can close cuda stub to passby"
#endif #endif


Loading…
Cancel
Save