Skip to content

Commit

Permalink
fix on
Browse files Browse the repository at this point in the history
  • Loading branch information
PaliC committed Jul 24, 2024
1 parent a162095 commit 63ff235
Showing 1 changed file with 13 additions and 13 deletions.
26 changes: 13 additions & 13 deletions .github/workflows/triton-inductor.yml
Original file line number Diff line number Diff line change
Expand Up @@ -26,16 +26,16 @@ jobs:
runs-on: "linux.12xlarge"
steps:
- name: Build
uses: pytorch/pytorch/.github/actions/linux-build@gh/PaliC/263/head
with:
build-environment: linux-jammy-py3.8-gcc11-build
docker-image-name: pytorch-linux-jammy-py3.8-gcc11-inductor-benchmarks
test-matrix: |
{ include: [
{ config: "cpu_inductor_huggingface", shard: 1, num_shards: 1 },
{ config: "cpu_inductor_timm", shard: 1, num_shards: 2 },
{ config: "cpu_inductor_timm", shard: 2, num_shards: 2 },
{ config: "cpu_inductor_torchbench", shard: 1, num_shards: 2 },
]}
secrets:
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
uses: pytorch/pytorch/.github/actions/linux-build@gh/PaliC/263/head
with:
build-environment: linux-jammy-py3.8-gcc11-build
docker-image-name: pytorch-linux-jammy-py3.8-gcc11-inductor-benchmarks
test-matrix: |
{ include: [
{ config: "cpu_inductor_huggingface", shard: 1, num_shards: 1 },
{ config: "cpu_inductor_timm", shard: 1, num_shards: 2 },
{ config: "cpu_inductor_timm", shard: 2, num_shards: 2 },
{ config: "cpu_inductor_torchbench", shard: 1, num_shards: 2 },
]}
secrets:
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}

0 comments on commit 63ff235

Please sign in to comment.