提交 3251f501 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: efa38f00d15518ed3622eab085a210c5018d782d
上级 2c2df830
此差异已折叠。
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册