From 68cb73950f0bed4bc6e9c657728a099a5dd1d4a5 Mon Sep 17 00:00:00 2001
From: mengfeil <mengfei.li@intel.com>
Date: Wed, 25 Dec 2024 14:34:29 +0800
Subject: [PATCH] apply pr for wheel

---
 .github/workflows/_linux_ut.yml            | 1 +
 .github/workflows/nightly_ondemand_whl.yml | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/_linux_ut.yml b/.github/workflows/_linux_ut.yml
index 65932365b..9b0310528 100644
--- a/.github/workflows/_linux_ut.yml
+++ b/.github/workflows/_linux_ut.yml
@@ -112,6 +112,7 @@ jobs:
             TORCH_COMMIT_ID=$(python -c 'import torch; print(torch.version.git_version)')
             cd ../pytorch
             git reset --hard && git checkout ${TORCH_COMMIT_ID}
+            python ../torch-xpu-ops/.github/scripts/apply_torch_pr.py
             TORCH_XPU_OPS_COMMIT=$(<third_party/xpu.txt)
             rm -rf third_party/torch-xpu-ops
             git clone https://github.com/intel/torch-xpu-ops.git third_party/torch-xpu-ops
diff --git a/.github/workflows/nightly_ondemand_whl.yml b/.github/workflows/nightly_ondemand_whl.yml
index 4815f15d6..a75a52625 100644
--- a/.github/workflows/nightly_ondemand_whl.yml
+++ b/.github/workflows/nightly_ondemand_whl.yml
@@ -113,7 +113,7 @@ jobs:
           cd pytorch && git checkout ${TORCH_COMMIT_ID}
           # apply PRs for stock pytorch
           pip install requests
-          # python ../torch-xpu-ops/.github/scripts/apply_torch_pr.py
+          python ../torch-xpu-ops/.github/scripts/apply_torch_pr.py
           git status && git show -s
           pip install -r requirements.txt
           TORCH_XPU_OPS_COMMIT=$(<third_party/xpu.txt)