cmake 版本检查
cmake_minimum_required (VERSION 3.18)
find_package 查找依赖
| 包 | 命令 |
|---|---|
| Python3 | find_package(Python3 REQUIRED COMPONENTS Development) |
| CUDA | find_package(CUDA 10.1 REQUIRED) |
| MPI | find_package(MPI REQUIRED) |
| NCCL | find_package(NCCL REQUIRED) |
| CUDAToolkit | find_package(CUDAToolkit REQUIRED) |
设置 option & 使用 option
设置 option
option(TRITON_ENABLE_GPU "Enable GPU support in backend" ON)
使用 option
if(${TRITON_ENABLE_GPU})
find_package(CUDAToolkit REQUIRED)
endif() # TRITON_ENABLE_GPU
编译选项
target_compile_features(target-name PRIVATE cxx_std_14) - 使用 gcc 的 cxx_std_14 选项