File tree Expand file tree Collapse file tree 1 file changed +13
-5
lines changed Expand file tree Collapse file tree 1 file changed +13
-5
lines changed Original file line number Diff line number Diff line change @@ -7,13 +7,21 @@ requires = [
7
7
build-backend = " setuptools.build_meta"
8
8
9
9
[tool .cibuildwheel ]
10
- environment.cuda = " 12.9"
11
- environment.temp = " /tmp/cuda-temp"
12
- environment.CUDA_PATH = " /usr/local/cuda-${cuda}"
13
- environment.PATH = " ${PATH}:${CUDA_PATH}/bin"
14
- environment.LD_LIBRARY_PATH =" ${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64"
15
10
xbuild-tools = [" cmake" , " ninja" ]
16
11
12
+ [tool .cibuildwheel .environment ]
13
+ cuda = " 12.9"
14
+ temp = " /tmp/cuda-temp"
15
+ CUDA_PATH = " /usr/local/cuda-${cuda}"
16
+ PATH = " ${PATH}:${CUDA_PATH}/bin"
17
+ LD_LIBRARY_PATH =" ${LD_LIBRARY_PATH}:${CUDA_PATH}/lib:${CUDA_PATH}/lib64"
18
+
19
+ # pyproject.toml
20
+ [tool .cibuildwheel .config-settings ]
21
+ "cmake.args" = [
22
+ " -DUSE_CUDA=ON"
23
+ ]
24
+
17
25
[tool .cibuildwheel .linux ]
18
26
before-all = [
19
27
" chmod +x tensor-array-repo/Tensor-Array/scripts/actions/install-cuda-rhel.sh" ,
You can’t perform that action at this time.
0 commit comments