From 44739dad93a50afb9bf31a7d8f6079fa9e3e3887 Mon Sep 17 00:00:00 2001 From: Paul Zhang Date: Wed, 6 Nov 2024 09:27:13 -0800 Subject: [PATCH] Update Dynamic Embedding Wheels CI (#2545) Summary: Pull Request resolved: https://github.com/pytorch/torchrec/pull/2545 Differential Revision: D65543488 --- .github/workflows/build_dynamic_embedding_wheels.yml | 4 ++-- contrib/dynamic_embedding/tools/before_linux_build.sh | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/build_dynamic_embedding_wheels.yml b/.github/workflows/build_dynamic_embedding_wheels.yml index 339e8e734..3abc051bc 100644 --- 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 index 9badc61d7..14cba7d0c 100755 --- 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}')