-
Notifications
You must be signed in to change notification settings - Fork 1
/
CMakeLists.txt
98 lines (79 loc) · 3.83 KB
/
CMakeLists.txt
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
include(CheckLanguage)
set(CMAKE_VERBOSE_MAKEFILE on)
cmake_minimum_required(VERSION 3.18)
if (NOT SKBUILD)
message(WARNING "\
This CMake file is meant to be executed using 'scikit-build'. Running
it directly will almost certainly not produce the desired result. If
you are a user trying to install this package, please use the command
below, which will install all necessary build dependencies, compile
the package in an isolated environment, and then install it.
=====================================================================
$ pip install .
=====================================================================
If you are a software developer, and this is your own package, then
it is usually much more efficient to install the build dependencies
in your environment once and use the following command that avoids
a costly creation of a new virtual environment at every compilation:
=====================================================================
$ pip install nanobind scikit-build-core[pyproject]
$ pip install --no-build-isolation -ve .
=====================================================================
You may optionally add -Ceditable.rebuild=true to auto-rebuild when
the package is imported. Otherwise, you need to re-run the above
after editing C++ files.")
endif()
check_language(CUDA)
if (CMAKE_CUDA_COMPILER)
message(STATUS "CUDA is OK")
include_directories(${CUDA_INCLUDE_DIRS})
set(CMAKE_CUDA_STANDARD 17)
set(CMAKE_CUDA_STANDARD_REQUIRED ON)
add_compile_definitions(USE_CUDA)
#set(CMAKE_CUDA_ARCHITECTURES 52)
project(${SKBUILD_PROJECT_NAME} LANGUAGES CXX CUDA)
else()
message(STATUS "No CUDA")
remove_definitions(USE_CUDA)
project(${SKBUILD_PROJECT_NAME} LANGUAGES CXX)
endif()
find_package(Python 3.8 COMPONENTS Interpreter Development.Module REQUIRED)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED True)
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/ext/nanobind)
find_package(nanobind CONFIG REQUIRED)
nanobind_add_module(
xgpr_cpu_rfgen_cpp_ext
# Target the stable ABI for Python 3.12+, which reduces
# the number of binary wheels that must be built. This
# does nothing on older Python versions
STABLE_ABI
NB_STATIC
xGPR/random_feature_generation/cpu_rf_gen/xgpr_cpu_rfgen_cpp_ext.cpp
xGPR/random_feature_generation/cpu_rf_gen/shared_fht_functions/hadamard_transforms.cpp
xGPR/random_feature_generation/cpu_rf_gen/shared_fht_functions/shared_rfgen_ops.cpp
xGPR/random_feature_generation/cpu_rf_gen/shared_fht_functions/simplex_rff_projections.cpp
xGPR/random_feature_generation/cpu_rf_gen/basic_ops/transform_functions.cpp
xGPR/random_feature_generation/cpu_rf_gen/rbf_ops/rbf_ops.cpp
xGPR/random_feature_generation/cpu_rf_gen/rbf_ops/ard_ops.cpp
xGPR/random_feature_generation/cpu_rf_gen/convolution_ops/conv1d_operations.cpp
xGPR/random_feature_generation/cpu_rf_gen/convolution_ops/rbf_convolution.cpp
)
if (CMAKE_CUDA_COMPILER)
nanobind_add_module(
xgpr_cuda_rfgen_cpp_ext
# Target the stable ABI for Python 3.12+, which reduces
# the number of binary wheels that must be built. This
# does nothing on older Python versions
STABLE_ABI
NB_STATIC
xGPR/random_feature_generation/gpu_rf_gen/xgpr_cuda_rfgen_cpp_ext.cpp
xGPR/random_feature_generation/gpu_rf_gen/basic_ops/basic_array_operations.cu
xGPR/random_feature_generation/gpu_rf_gen/rbf_ops/rbf_ops.cu
xGPR/random_feature_generation/gpu_rf_gen/rbf_ops/ard_ops.cu
xGPR/random_feature_generation/gpu_rf_gen/convolution_ops/convolution.cu
xGPR/random_feature_generation/gpu_rf_gen/convolution_ops/rbf_convolution.cu
)
install(TARGETS xgpr_cuda_rfgen_cpp_ext LIBRARY DESTINATION ${SKBUILD_PROJECT_NAME})
endif()
install(TARGETS xgpr_cpu_rfgen_cpp_ext LIBRARY DESTINATION ${SKBUILD_PROJECT_NAME})