Skip to content

Commit

Permalink
tmp
Browse files Browse the repository at this point in the history
  • Loading branch information
xyang16 committed Apr 30, 2024
1 parent 12aa4fc commit 29b3e0a
Showing 1 changed file with 2 additions and 121 deletions.
123 changes: 2 additions & 121 deletions .github/workflows/nightly_publish.yml
Original file line number Diff line number Diff line change
Expand Up @@ -48,129 +48,10 @@ jobs:
${{ github.workspace }}/**/build/reports/**/*
!${{ github.workspace }}/**/build/reports/jacoco/*
test-pytorch:
if: github.repository == 'deepjavalibrary/djl'
runs-on: ${{ matrix.operating-system }}
strategy:
matrix:
operating-system: [ macos-latest, ubuntu-latest, windows-latest ]

steps:
- uses: actions/checkout@v4
- name: Set up JDK 17
uses: actions/setup-java@v4
with:
distribution: 'corretto'
java-version: 17
# Enable gradle cache: https://github.com/actions/cache/blob/master/examples.md#java---gradle
- uses: actions/cache@v4
with:
path: ~/.gradle/caches
key: ${{ runner.os }}-gradle-${{ hashFiles('**/*.gradle*') }}
restore-keys: |
${{ runner.os }}-gradle-
- name: Test with Gradle
run: ./gradlew :integration:test "-Dai.djl.default_engine=PyTorch"

test-tensorflow:
if: github.repository == 'deepjavalibrary/djl'
runs-on: ${{ matrix.operating-system }}
strategy:
matrix:
operating-system: [ macos-latest, ubuntu-latest, windows-latest ]

steps:
- uses: actions/checkout@v4
- name: Set up JDK 17
uses: actions/setup-java@v4
with:
distribution: 'corretto'
java-version: 17
# Enable gradle cache: https://github.com/actions/cache/blob/master/examples.md#java---gradle
- uses: actions/cache@v4
with:
path: ~/.gradle/caches
key: ${{ runner.os }}-gradle-${{ hashFiles('**/*.gradle*') }}
restore-keys: |
${{ runner.os }}-gradle-
- name: Test with Gradle
run: ./gradlew :integration:test "-Dai.djl.default_engine=TensorFlow"

test-aarch64:
if: github.repository == 'deepjavalibrary/djl'
runs-on: [ self-hosted, aarch64 ]
container: amazonlinux:2
timeout-minutes: 30
needs: create-runners
steps:
- name: Setup Environment
run: |
yum -y update
yum install -y tar gzip
# checkout@v4 requires GLIBC 2.27
- uses: actions/checkout@v3
- name: Set up JDK 17
uses: actions/setup-java@v3
with:
java-version: 17
distribution: corretto
architecture: aarch64
- uses: actions/cache@v3
with:
path: ~/.gradle/caches
key: ${{ runner.os }}-gradle-${{ hashFiles('**/*.gradle*') }}
restore-keys: |
${{ runner.os }}-gradle-
- name: Test with Gradle
run: |
./gradlew :integration:test "-Dai.djl.default_engine=PyTorch"
./gradlew :integration:clean
# OnnxRuntime 1.17.1 requires GLIBC 2.27
# ./gradlew :integration:test "-Dai.djl.default_engine=OnnxRuntime"
# ./gradlew :integration:clean
test-cuda-121:
if: github.repository == 'deepjavalibrary/djl'
runs-on: [ self-hosted, gpu ]
container:
image: nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu20.04
options: --gpus all --runtime=nvidia
timeout-minutes: 30
needs: create-runners
steps:
- name: Setup Environment
run: |
apt-get update
apt-get install -y software-properties-common wget locales libfontconfig1
locale-gen en_US.UTF-8
- uses: actions/checkout@v4
- name: Set up JDK 17
uses: actions/setup-java@v4
with:
java-version: 17
distribution: corretto
- uses: actions/cache@v4
with:
path: ~/.gradle/caches
key: ${{ runner.os }}-gradle-${{ hashFiles('**/*.gradle*') }}
restore-keys: |
${{ runner.os }}-gradle-
- name: Test with Gradle
run: |
./gradlew :integration:test
./gradlew :engines:tensorflow:tensorflow-model-zoo:test
./gradlew :integration:clean :integration:test -Dai.djl.default_engine=PyTorch
./gradlew :integration:clean :integration:test -Dai.djl.default_engine=TensorFlow
./gradlew :engines:mxnet:mxnet-model-zoo:test
./gradlew :engines:pytorch:pytorch-model-zoo:test
./gradlew :engines:tensorflow:tensorflow-model-zoo:test
./gradlew :engines:tensorrt:test
./gradlew :engines:onnxruntime:onnxruntime-engine:test
publish:
if: github.repository == 'deepjavalibrary/djl'
runs-on: ubuntu-latest
needs: [ build, test-pytorch, test-tensorflow, test-aarch64, test-cuda-121 ]
needs: [ build ]
steps:
- uses: actions/checkout@v4
- name: Set up JDK 17
Expand Down Expand Up @@ -249,7 +130,7 @@ jobs:
stop-runners:
if: ${{ github.repository == 'deepjavalibrary/djl' && always() }}
runs-on: [ self-hosted, scheduler ]
needs: [ create-runners, test-aarch64, test-cuda-121 ]
needs: [ create-runners ]
steps:
- name: Stop all instances
run: |
Expand Down

0 comments on commit 29b3e0a

Please sign in to comment.