Skip to content

Commit 0a84a99

Browse files
Update pyproject.toml
1 parent 5de8d6d commit 0a84a99

File tree

1 file changed

+13
-5
lines changed

1 file changed

+13
-5
lines changed

pyproject.toml

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,21 @@ requires = [
77
build-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"
1510
xbuild-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]
1826
before-all = [
1927
"chmod +x tensor-array-repo/Tensor-Array/scripts/actions/install-cuda-rhel.sh",

0 commit comments

Comments
 (0)