Skip to content

Commit

Permalink
[pytorch] support pytorch 2.1.2 for Neuron
Browse files Browse the repository at this point in the history
  • Loading branch information
siddvenk committed Nov 20, 2024
1 parent 1db031b commit 3bce796
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion .github/workflows/nightly_publish.yml
Original file line number Diff line number Diff line change
Expand Up @@ -176,10 +176,12 @@ jobs:
- name: Publish to snapshot repository
if: ${{ github.event.inputs.mode == '' || github.event.inputs.mode == 'snapshot' }}
# PT 1.13.1 is for PT 1.x support, and CUDA 11.x support
# PT 2.1.2 is for Neuron support (latest supported version on neuron)
# PT 2.3.1 is for CUDA 12.1 support
# PT 2.5.1 is for CUDA 12.4 support
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.3.1 -Psnapshot
./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.5.1 -Psnapshot
./gradlew clean engines:ml:xgboost:publish -Pgpu -Psnapshot
Expand All @@ -197,7 +199,6 @@ jobs:
./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.3.1 -P${{ github.event.inputs.mode }}
./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.4.0 -P${{ github.event.inputs.mode }}
./gradlew clean engines:pytorch:pytorch-jni:publish -Ppt_version=2.5.1 -P${{ github.event.inputs.mode }}
./gradlew clean engines:ml:xgboost:publish -Pgpu -P${{ github.event.inputs.mode }}
./gradlew clean publish -P${{ github.event.inputs.mode }}
Expand Down
2 changes: 1 addition & 1 deletion engines/pytorch/pytorch-engine/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ The following table illustrates which pytorch version that DJL supports:

| PyTorch engine version | PyTorch native library version |
|------------------------|-------------------------------------------|
| pytorch-engine:0.32.0 | 1.13.1, 2.3.1, **2.5.1** |
| pytorch-engine:0.32.0 | 1.13.1, 2.1.2, 2.3.1, **2.5.1** |
| pytorch-engine:0.31.0 | 1.13.1, 2.1.2, 2.3.1, 2.4.0, **2.5.1** |
| pytorch-engine:0.30.0 | 1.13.1, 2.1.2, 2.3.1, **2.4.0** |
| pytorch-engine:0.29.0 | 1.13.1, 2.1.2, 2.2.2, **2.3.1** |
Expand Down

0 comments on commit 3bce796

Please sign in to comment.