Browse Source

feat(dnn): differentiate sass kernel with cuda version

GitOrigin-RevId: 40bb4423b8
tags/v1.9.0
Megvii Engine Team 3 years ago
parent
commit
286051ede1
3 changed files with 2 additions and 1 deletions
  1. +0
    -1
      .gitattributes
  2. +1
    -0
      CMakeLists.txt
  3. +1
    -0
      dnn/src/CMakeLists.txt

+ 0
- 1
.gitattributes View File

@@ -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


+ 1
- 0
CMakeLists.txt View File

@@ -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)


+ 1
- 0
dnn/src/CMakeLists.txt View File

@@ -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)


Loading…
Cancel
Save