inductor-periodic #611
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: inductor-periodic | |
on: | |
push: | |
tags: | |
- ciflow/inductor/* | |
workflow_dispatch: | |
schedule: | |
# Run every 4 hours during the week and every 12 hours on the weekend | |
- cron: 45 0,4,8,12,16,20 * * 1-5 | |
- cron: 45 4,12 * * 0,6 | |
concurrency: | |
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ github.ref_type == 'branch' && github.sha }}-${{ github.event_name == 'workflow_dispatch' }} | |
cancel-in-progress: true | |
jobs: | |
linux-focal-cuda12_1-py3_10-gcc9-periodic-dynamo-benchmarks-build: | |
name: cuda12.1-py3.10-gcc9-sm86-periodic-dynamo-benchmarks | |
uses: ./.github/workflows/_linux-build.yml | |
with: | |
build-environment: linux-focal-cuda12.1-py3.10-gcc9-sm86 | |
docker-image-name: pytorch-linux-focal-cuda12.1-cudnn8-py3-gcc9-inductor-benchmarks | |
cuda-arch-list: '8.6' | |
test-matrix: | | |
{ include: [ | |
{ config: "dynamo_eager_torchbench", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "dynamo_eager_huggingface", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "dynamo_eager_timm", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "dynamo_eager_timm", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "aot_eager_torchbench", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "aot_eager_huggingface", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "aot_eager_timm", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "aot_eager_timm", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "dynamic_aot_eager_torchbench", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "dynamic_aot_eager_huggingface", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "dynamic_aot_eager_timm", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
{ config: "dynamic_aot_eager_timm", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" }, | |
]} | |
secrets: | |
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} | |
linux-focal-cuda12_1-py3_10-gcc9-periodic-dynamo-benchmarks-test: | |
name: cuda12.1-py3.10-gcc9-sm86-periodic-dynamo-benchmarks | |
uses: ./.github/workflows/_linux-test.yml | |
needs: linux-focal-cuda12_1-py3_10-gcc9-periodic-dynamo-benchmarks-build | |
with: | |
build-environment: linux-focal-cuda12.1-py3.10-gcc9-sm86 | |
docker-image: ${{ needs.linux-focal-cuda12_1-py3_10-gcc9-periodic-dynamo-benchmarks-build.outputs.docker-image }} | |
test-matrix: ${{ needs.linux-focal-cuda12_1-py3_10-gcc9-periodic-dynamo-benchmarks-build.outputs.test-matrix }} | |
secrets: | |
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }} |