Llama-diffusion/CMakeLists.txt at main · luozixin2/Llama-diffusion · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
cmake_minimum_required(VERSION 3.18)

# 设置 MSVC 运行时（Windows专用，Linux下忽略但保留无妨）
set(CMAKE_MSVC_RUNTIME_LIBRARY "MultiThreaded$<$<CONFIG:Debug>:Debug>")

set(MY_CUDA_ROOT "/home/lzx/cuda-12.8")

# 1. 告诉 CMake 哪里找 CUDA 工具包 (用于 find_package)
set(CUDAToolkit_ROOT "${MY_CUDA_ROOT}" CACHE PATH "CUDA Toolkit Root" FORCE)

# 2. 强制指定 nvcc 编译器路径
set(CMAKE_CUDA_COMPILER "${MY_CUDA_ROOT}/bin/nvcc")

# 3. 辅助变量，防止 CMake 查找系统默认路径
set(CUDA_TOOLKIT_ROOT_DIR "${MY_CUDA_ROOT}" CACHE PATH "Legacy CUDA Root" FORCE)

project(SDAR_LlamaDiffusionProject LANGUAGES C CXX)

# C++17
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

# ==========================================
# 全局开启 -fPIC
# ==========================================
set(CMAKE_POSITION_INDEPENDENT_CODE ON)

# 强制静态链接子模块，方便打包进 Python 扩展
set(BUILD_SHARED_LIBS OFF CACHE BOOL "Build static libraries" FORCE)

# GPU 选项
option(LLAMA_USE_ACCELERATE "Use Apple Accelerate (macOS)" OFF)
option(GGML_CUDA       "Use cuBLAS (NVIDIA GPUs)"   ON)
option(LLAMA_METAL        "Use Apple Metal (macOS)"     OFF)

# 如果开启了 CUDA，显式启用 CUDA 语言支持
if(GGML_CUDA)
    enable_language(CUDA)
endif()

# ---------------------------------------------------
# 1) extern/pybind11
add_subdirectory(extern/pybind11)

# 2) extern/llama.cpp
# 防止 llama.cpp 构建测试和示例，加快编译速度
set(LLAMA_BUILD_TESTS OFF CACHE BOOL "" FORCE)
set(LLAMA_BUILD_EXAMPLES OFF CACHE BOOL "" FORCE)
set(LLAMA_BUILD_SERVER OFF CACHE BOOL "" FORCE)

add_subdirectory(extern/llama.cpp)

# ==========================================
# 双重保险：强制 llama 目标使用 PIC
# ==========================================
if(TARGET llama)
    set_property(TARGET llama PROPERTY POSITION_INDEPENDENT_CODE ON)
endif()
if(TARGET ggml)
    set_property(TARGET ggml PROPERTY POSITION_INDEPENDENT_CODE ON)
endif()
if(TARGET ggml-base) # 新版 llama.cpp 可能还有这个
    set_property(TARGET ggml-base PROPERTY POSITION_INDEPENDENT_CODE ON)
endif()
if(TARGET ggml-cuda) # 如果有独立的 cuda 目标
    set_property(TARGET ggml-cuda PROPERTY POSITION_INDEPENDENT_CODE ON)
endif()

# 3) llama_diffusion
add_subdirectory(llama_diffusion)