Skip to content

Commit 44739da

Browse files
PaulZhang12facebook-github-bot
authored andcommitted
Update Dynamic Embedding Wheels CI (#2545)
Summary: Pull Request resolved: #2545 Differential Revision: D65543488
1 parent 509b0d2 commit 44739da

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.github/workflows/build_dynamic_embedding_wheels.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@ jobs:
2020
fail-fast: false
2121
matrix:
2222
os: [ ubuntu-latest ]
23-
pyver: [ cp39, cp310 ]
24-
cuver: [ "11.8" ]
23+
pyver: [ cp39, cp310, cp311, cp312 ]
24+
cuver: [ "12.1" ]
2525

2626
steps:
2727
-

contrib/dynamic_embedding/tools/before_linux_build.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ set -xe
99

1010
distro=rhel7
1111
arch=x86_64
12-
CUDA_VERSION="${CUDA_VERSION:-11.8}"
12+
CUDA_VERSION="${CUDA_VERSION:-12.1}"
1313

1414
CUDA_MAJOR_VERSION=$(echo "${CUDA_VERSION}" | tr '.' ' ' | awk '{print $1}')
1515
CUDA_MINOR_VERSION=$(echo "${CUDA_VERSION}" | tr '.' ' ' | awk '{print $2}')

0 commit comments

Comments
 (0)