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 = [
77build-backend = " setuptools.build_meta"
88
99[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"
1510xbuild-tools = [" cmake" , " ninja" ]
1611
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+
1725[tool .cibuildwheel .linux ]
1826before-all = [
1927 " 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