shijin/cuElim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加了指定nvcc路径以及主机编译器的指令

Browse Source
This commit is contained in:
shijin 2024-11-04 14:46:18 +08:00
parent 2ef7ce5844
commit 21acd97bdc
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
CMakeLists.txt
View File

@ -1,15 +1,21 @@
cmake_minimum_required(VERSION 3.24) # 设置cmake最低版本 cmake_minimum_required(VERSION 3.24) # 设置cmake最低版本
# 设置 nvcc 的路径
set(CMAKE_CUDA_COMPILER /usr/local/cuda/bin/nvcc) # 根据实际安装路径设置
# 设置 nvcc 使用的主机编译器
set(CMAKE_CUDA_HOST_COMPILER /usr/bin/g++-11)
# 设置 C++ 标准 # 设置 C++ 标准
set(CMAKE_CXX_STANDARD 20) # set(CMAKE_CXX_STANDARD 20)
set(CMAKE_CXX_STANDARD_REQUIRED True) # set(CMAKE_CXX_STANDARD_REQUIRED True)
# 设置 CUDA C++ 标准 # 设置 CUDA C++ 标准
set(CMAKE_CUDA_STANDARD 20) set(CMAKE_CUDA_STANDARD 20)
set(CMAKE_CUDA_STANDARD_REQUIRED True) set(CMAKE_CUDA_STANDARD_REQUIRED True)
set(CMAKE_CUDA_ARCHITECTURES native) # 设置CUDA架构 set(CMAKE_CUDA_ARCHITECTURES native) # 设置CUDA架构
# 设置编译选项 # 设置编译选项
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3") # set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3")
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -O3 -maxrregcount=128") set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -O3 -maxrregcount=128")
project(cuElim LANGUAGES CXX CUDA) # 设置项目名和语言 project(cuElim LANGUAGES CXX CUDA) # 设置项目名和语言