From 286051ede10bdb3603a0873f1bf5ed5f94feac16 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 1 Mar 2022 19:21:10 +0800
Subject: [PATCH] feat(dnn): differentiate sass kernel with cuda version

GitOrigin-RevId: 40bb4423b86ec5d0016cffe405ba51c66d502f16
---
 .gitattributes         | 1 -
 CMakeLists.txt         | 1 +
 dnn/src/CMakeLists.txt | 1 +
 3 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.gitattributes b/.gitattributes
index 0b84a4c2c..14b131c07 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -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
diff --git a/CMakeLists.txt b/CMakeLists.txt
index fe1f2876d..565917762 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -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)
diff --git a/dnn/src/CMakeLists.txt b/dnn/src/CMakeLists.txt
index 2113839cc..9db596c3c 100644
--- a/dnn/src/CMakeLists.txt
+++ b/dnn/src/CMakeLists.txt
@@ -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)
-- 
GitLab