From 540ad49ae908b8ec2a6f30b4461edf9cf1972d8a Mon Sep 17 00:00:00 2001 From: Frank Liu Date: Thu, 27 Jun 2024 18:11:02 -0700 Subject: [PATCH] [pytorch] Adds back PyTorch 2.1.2 support --- .github/workflows/nightly_publish.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/nightly_publish.yml b/.github/workflows/nightly_publish.yml index cb41e9a609b..aa0f324e308 100644 --- a/.github/workflows/nightly_publish.yml +++ b/.github/workflows/nightly_publish.yml @@ -189,6 +189,7 @@ jobs: if: ${{ github.event.inputs.mode == '' || github.event.inputs.mode == 'snapshot' }} run: | ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=1.13.1 -Psnapshot + ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.1.2 -Psnapshot ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.2.2 -Psnapshot ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.3.1 -Psnapshot ./gradlew clean engines:ml:xgboost:publish -Pgpu -Psnapshot @@ -204,6 +205,7 @@ jobs: if: ${{ github.event.inputs.mode == 'staging' }} run: | ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=1.13.1 -P${{ github.event.inputs.mode }} + ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.1.2 -P${{ github.event.inputs.mode }} ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.2.2 -P${{ github.event.inputs.mode }} ./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.3.1 -P${{ github.event.inputs.mode }} ./gradlew clean engines:ml:xgboost:publish -Pgpu -P${{ github.event.inputs.mode }}