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

GitOrigin-RevId: efa38f00d15518ed3622eab085a210c5018d782d

fix(mgb/cuda-stub): add libcuda-wrap_11.4.h to fit the CUDA11.4 toolchain
GitOrigin-RevId: efa38f00d15518ed3622eab085a210c5018d782d
3251f501 · Megvii Engine Team · 2c2df830 · 3251f501 · 3251f501
展开全部隐藏空白更改
内联并排

Showing with 7840 addition and 0 deletion

dnn/cuda-stub/src/libcuda-wrap_11.4.h dnn/cuda-stub/src/libcuda-wrap_11.4.h +7838 -0

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

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