Setting up IPEX for benchmarks depending on the USE_IPEX environment …

…variable (#2320) Hot fix for https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/11003865037 --------- Signed-off-by: Anatoly Myachev <[email protected]>
intel · Sep 24, 2024 · a164fe6 · a164fe6
1 parent f2b4b6d
commit a164fe6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/triton-benchmarks.yml b/.github/workflows/triton-benchmarks.yml
@@ -55,19 +55,19 @@ jobs:
           pip install wheel cmake
 
       - name: Setup PyTorch with IPEX
-        if: ${{ github.event_name == 'schedule' || inputs.install_ipex }}
+        if: ${{ env.USE_IPEX == '1' }}
         uses: ./.github/actions/setup-pytorch
         with:
           repository: Stonepia/pytorch
 
       - name: Setup PyTorch without IPEX
-        if: ${{ !(github.event_name == 'schedule' || inputs.install_ipex) }}
+        if: ${{ env.USE_IPEX == '0' }}
         uses: ./.github/actions/setup-pytorch
         with:
           repository: pytorch/pytorch
 
       - name: Setup IPEX
-        if: ${{ github.event_name == 'schedule' || inputs.install_ipex }}
+        if: ${{ env.USE_IPEX == '1' }}
         uses: ./.github/actions/setup-ipex
 
       - name: Build Triton wheels