canary-al2.yml

name: Canary-CentOS

on:
  workflow_call:
    inputs:
      repo-id:
        description: 'staging repository id to test'
        required: false
        type: string
        default: ''
      djl-version:
        description: 'djl version to test'
        type: string
        required: false
      pt-version:
        description: 'pytorch version to test'
        type: string
        required: false
        default: ''
  workflow_dispatch:
    inputs:
      repo-id:
        description: 'staging repository id to test'
        required: false
        default: ''
      djl-version:
        description: 'djl version to test'
        required: false
      pt-version:
        description: 'pytorch version to test'
        required: false
        default: ''

jobs:
  canary-test-al2:
    if: github.repository == 'deepjavalibrary/djl-demo'
    runs-on: ubuntu-latest
    env:
      AWS_REGION: us-east-1
      DJL_STAGING: ${{github.event.inputs.repo-id}}
      DJL_VERSION: ${{github.event.inputs.djl-version}}
      PT_VERSION: ${{github.event.inputs.pt-version}}
      JAVA_HOME: /usr/lib/jvm/java-17-amazon-corretto
    container: amazonlinux:2
    steps:
      - name: Install libgomp
        run: |
          yum install -y tar gzip libgomp
          yum -y install java-17-amazon-corretto-devel
      - uses: taiki-e/checkout-action@v1
      - name: Test PyTorch
        working-directory: canary
        run: |
          set -x
          DJL_ENGINE=pytorch-native-auto PYTORCH_VERSION=$PT_VERSION PYTORCH_PRECXX11=true ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-auto PYTORCH_PRECXX11=true PYTORCH_VERSION=1.13.1 ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-auto PYTORCH_PRECXX11=true PYTORCH_VERSION=2.1.2 ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-auto PYTORCH_PRECXX11=true PYTORCH_VERSION=2.3.1 ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-auto PYTORCH_PRECXX11=true PYTORCH_VERSION=2.4.0 ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-cpu-precxx11 ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-cpu-precxx11 PT_VERSION=1.13.1 ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-cpu-precxx11 PT_VERSION=2.1.2 ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-cpu-precxx11 PT_VERSION=2.3.1 ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=pytorch-native-cpu-precxx11 PT_VERSION=2.4.0 ./gradlew clean run
          rm -rf /root/.djl.ai/
      - name: Test TensorFlow
        working-directory: canary
        run: |
          set -x
          DJL_ENGINE=tensorflow-native-auto ./gradlew clean run
          rm -rf /root/.djl.ai/
          DJL_ENGINE=tensorflow-native-cpu ./gradlew clean run
          rm -rf /root/.djl.ai/
      # OnnxRuntime 1.17.1 requires GLIBC 2.27
      # - name: Test OnnxRuntime
      #   working-directory: canary
      #   run: |
      #     set -x
      #     DJL_ENGINE=onnxruntime PYTORCH_PRECXX11=true ./gradlew clean run
      #     rm -rf /root/.djl.ai/
      - name: Test Xgboost
        working-directory: canary
        run: |
          set -x
          DJL_ENGINE=xgboost PYTORCH_PRECXX11=true ./gradlew clean run
          rm -rf /root/.djl.ai/
      - name: Test LightGBM
        working-directory: canary
        run: |
          set -x
          DJL_ENGINE=lightgbm PYTORCH_PRECXX11=true ./gradlew clean run
          rm -rf /root/.djl.ai/
      - name: Test Huggingface tokenizers
        working-directory: canary
        run: |
          set -x
          DJL_ENGINE=tokenizers PYTORCH_PRECXX11=true ./gradlew clean run
          rm -rf /root/.djl.ai/
      - name: Test SentencePiece tokenizers
        working-directory: canary
        run: |
          set -x
          DJL_ENGINE=sentencepiece PYTORCH_PRECXX11=true ./gradlew clean run
          rm -rf /root/.djl.ai/
      # MXNet doesn't support amazonlinux:2
      # fastText doesn't support amazonlinux:2