From 3a1df743524a30c1d5945256578388018a3f768e Mon Sep 17 00:00:00 2001
From: "Ding, Yi1" <yi1.ding@intel.com>
Date: Mon, 23 Dec 2024 02:59:11 +0000
Subject: [PATCH] update

---
 .github/workflows/nightly_ondemand.yml | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/nightly_ondemand.yml b/.github/workflows/nightly_ondemand.yml
index ff743f384..476456fa1 100644
--- a/.github/workflows/nightly_ondemand.yml
+++ b/.github/workflows/nightly_ondemand.yml
@@ -8,6 +8,11 @@ on:
     - cron: '0 16 * * 5'
   workflow_dispatch:
     inputs:
+      pytorch_repo:
+        required: false
+        type: string
+        default: 'https://github.com/pytorch/pytorch.git'
+        description: Pytorch repo
       pytorch:
         required: false
         type: string
@@ -107,8 +112,8 @@ jobs:
           pwd
           cd ../ && rm -rf pytorch
           source activate e2e_ci
-          git clone https://github.com/DDEle/pytorch
-          cd pytorch && git checkout onednn_graph_sdpa
+          git clone ${{ inputs.pytorch_repo }} pytorch
+          cd pytorch && git checkout $(echo ${{ env.pytorch }} |awk '{print $1}') 
           # apply PRs for stock pytorch
           pip install requests
           pip install botocore