native_jni_s3_pytorch_android.yml

name: Native JNI S3 PyTorch Android

on:
  workflow_dispatch:

jobs:
  build-pytorch-jni-android:
    runs-on: ubuntu-18.04
    env:
      NDK_VERSION: "20.0.5594570"

    steps:
      - uses: actions/checkout@v2
      - name: Set up JDK 1.8
        uses: actions/setup-java@v1
        with:
          java-version: 1.8
      - uses: actions/cache@v2
        with:
          path: ~/.gradle/caches
          key: ${{ runner.os }}-gradle-${{ hashFiles('**/*.gradle*') }}
          restore-keys: |
            ${{ runner.os }}-gradle-
      - name: Install NDK
        run: echo "y" | sudo ${ANDROID_HOME}/tools/bin/sdkmanager --install "ndk;${NDK_VERSION}"
      - name: build android
        run: |
          export ANDROID_NDK=${ANDROID_SDK_ROOT}/ndk/${NDK_VERSION}
          ./gradlew :engines:pytorch:pytorch-native:compileAndroidJNI
      - name: Upload compiled jni library
        uses: actions/upload-artifact@v2
        if: always()
        with:
          name: jnilib-Android
          path: engines/pytorch/pytorch-native/jnilib

  publish:
    runs-on: ubuntu-18.04
    needs: [build-pytorch-jni-android]
    steps:
      - uses: actions/checkout@v2
      - name: Download compiledJNI Android
        uses: actions/download-artifact@v1
        with:
          name: jnilib-Android
          path: jnilib
      - name: Configure AWS Credentials
        uses: aws-actions/configure-aws-credentials@v1
        with:
          aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }}
          aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
          aws-region: us-east-2
      - name: Copy files to S3 with the AWS CLI
        run: |
          PYTORCH_VERSION="$(cat gradle.properties | awk -F '=' '/pytorch_version/ {print $2}')"
          aws s3 sync jnilib s3://djl-ai/publish/pytorch-${PYTORCH_VERSION}/jnilib
          aws cloudfront create-invalidation --distribution-id E371VB8JQ6NRVY --paths "/pytorch-${PYTORCH_VERSION}/jnilib*"