Skip to content

BetterTransformer GPU / Python - Test #172

BetterTransformer GPU / Python - Test

BetterTransformer GPU / Python - Test #172

name: BetterTransformer GPU / Python - Test
on:
workflow_dispatch:
schedule:
- cron: 0 1 */3 * * # at 1am every 3 days
jobs:
do-the-job:
name: Start self-hosted EC2 runner
runs-on: [single-gpu, nvidia-gpu, t4, ci]
env:
AWS_REGION: us-east-1
steps:
- name: Checkout
uses: actions/checkout@v2
- name: Build image
run: |
docker build -f tests/bettertransformer/Dockerfile_bettertransformer_gpu -t bettertransformer-gpu .
- name: Test with unittest within docker container
run: |
docker run --rm --gpus all --workdir=/workspace/optimum/tests bettertransformer-gpu:latest