提交 286051ed 编写于 作者: M Megvii Engine Team

feat(dnn): differentiate sass kernel with cuda version

GitOrigin-RevId: 40bb4423b86ec5d0016cffe405ba51c66d502f16
上级 f78b60ec
......@@ -6,7 +6,6 @@ dnn/src/cuda/conv_bias/int8/kimpl/* binary
dnn/src/cuda/conv_bias/int8_imma/kimpl/* binary
dnn/src/cuda/batch_conv_bias/int8/kimpl/* binary
dnn/src/cuda/matrix_mul/fp32_simt/kimpl/* binary
dnn/src/cuda/sass/prebuilt/map_defs.cpp binary
dnn/src/cuda/convolution/backward_data/int8/kimpl/* binary
dnn/src/cuda/elemwise_multi_type/kimpl/* binary
src/gopt/test/cache_data.h binary
......
......@@ -564,6 +564,7 @@ if(MGE_WITH_CUDA)
message(WARNING "force set CMAKE_CUDA_COMPILER to nvcc when nvcc sbsa mode!!")
endif()
find_package(CUDA)
enable_language(CUDA)
set(CMAKE_CUDA_STANDARD 14)
set(CMAKE_CUDA_STANDARD_REQUIRED ON)
......
......@@ -118,6 +118,7 @@ endif()
if(MGE_WITH_CUDA)
file(GLOB_RECURSE SOURCES_ cuda/*.cpp)
list(APPEND SOURCES ${SOURCES_})
file(GLOB_RECURSE CUSOURCES cuda/*.cu)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册