Update Dynamic Embedding Wheels CI (#2545)

Summary: Pull Request resolved: #2545 Differential Revision: D65543488
pytorch · Nov 6, 2024 · 44739da · 44739da
1 parent 509b0d2
commit 44739da
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/build_dynamic_embedding_wheels.yml b/.github/workflows/build_dynamic_embedding_wheels.yml
@@ -20,8 +20,8 @@ jobs:
       fail-fast: false
       matrix:
         os: [ ubuntu-latest ]
-        pyver: [ cp39, cp310 ]
-        cuver: [ "11.8" ]
+        pyver: [ cp39, cp310, cp311, cp312 ]
+        cuver: [ "12.1" ]
 
     steps:
       -

diff --git a/contrib/dynamic_embedding/tools/before_linux_build.sh b/contrib/dynamic_embedding/tools/before_linux_build.sh
@@ -9,7 +9,7 @@ set -xe
 
 distro=rhel7
 arch=x86_64
-CUDA_VERSION="${CUDA_VERSION:-11.8}"
+CUDA_VERSION="${CUDA_VERSION:-12.1}"
 
 CUDA_MAJOR_VERSION=$(echo "${CUDA_VERSION}" | tr '.' ' ' | awk '{print $1}')
 CUDA_MINOR_VERSION=$(echo "${CUDA_VERSION}" | tr '.' ' ' | awk '{print $2}')