From 0b4123eabcdf8caf6c469935c8f0944e4930b017 Mon Sep 17 00:00:00 2001 From: Anatoly Myachev Date: Wed, 11 Sep 2024 12:21:46 +0000 Subject: [PATCH] Provide the ability to run benchmarks with or without IPEX Signed-off-by: Anatoly Myachev --- .github/workflows/triton-benchmarks.yml | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/.github/workflows/triton-benchmarks.yml b/.github/workflows/triton-benchmarks.yml index 0c4647edda..0bc8800435 100644 --- a/.github/workflows/triton-benchmarks.yml +++ b/.github/workflows/triton-benchmarks.yml @@ -11,6 +11,10 @@ on: description: Tag for benchmark results type: string default: "test" + install_ipex: + description: Install Intel PyTorch Extension + type: boolean + default: true schedule: - cron: "5 23 * * *" @@ -49,10 +53,20 @@ jobs: run: | pip install wheel - - name: Setup PyTorch + - name: Setup PyTorch with IPEX + if: ${{ inputs.install_ipex }} uses: ./.github/actions/setup-pytorch + with: + repository: Stonepia/pytorch + + - name: Setup PyTorch without IPEX + if: ${{ !inputs.install_ipex }} + uses: ./.github/actions/setup-pytorch + with: + repository: pytorch/pytorch - name: Setup IPEX + if: ${{ inputs.install_ipex }} uses: ./.github/actions/setup-ipex - name: Build Triton wheels