error: invalid device function

原创

algorithm小白菜 2021-07-16 14:57:01 ©著作权

文章标签 其他 文章分类 代码人生

©著作权归作者所有：来自51CTO博客作者algorithm小白菜的原创作品，请联系作者获取转载授权，否则将追究法律责任

这个问题是最进最做科研时遇到的：他的原因就是gpu的配置问题
项目名 error: invalid device function
没图片了，只上解决方法吧：把总的cmakelist中的list处换成
list(APPEND CUDA_NVCC_FLAGS “-gencode;arch=compute_20,code=sm_20;-gencode;arch=compute_20,code=sm_21;-gencode;arch=compute_30,code=sm_30;-gencode;arch=compute_35,code=sm_35;-#gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_52,code=sm_52;-gencode;arch=compute_61,code=sm_61”)
其实就是在最后加了一句：-gencode;arch=compute_61,code=sm_61
最后一句需要你自己知道gpu的计算能力去具体配置

我自己的配置时是这样的：这个是在总的CMakeLists.txt中写的
SET(HAVE_CUDA 1)
LIST(APPEND CUDA_NVCC_FLAGS
"-gencode;arch=compute_50,code=sm_50;
–ftz=true;
–prec-div=false;
–prec-sqrt=false;
–expt-relaxed-constexpr"
)

if(UNIX OR APPLE)
LIST(APPEND CUDA_NVCC_FLAGS “-Xcompiler;-fPIC”)
endif()
加黑部分就是我自己根据自己的GPU计算能力来添加的，我觉得从工程复用的角度来看，在cmake中做好还是写最上面的那种配置，这样就不用去查看自己gpu的计算能力