From e04867997e8903b9f44b75d073ef0be8c3159c12 Mon Sep 17 00:00:00 2001
From: Disty0 <semihgulec2005@gmail.com>
Date: Fri, 26 May 2023 00:13:56 +0300
Subject: [PATCH] Use ipexrun when using --use-ipex

---
 webui.sh | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/webui.sh b/webui.sh
index ff9c58b4d..2119e4890 100755
--- a/webui.sh
+++ b/webui.sh
@@ -81,10 +81,20 @@ else
     exit 1
 fi
 
+if [[ "$@" == *"--use-ipex"* ]]
+then
+    echo "Setting OneAPI enviroment"
+    source /opt/intel/oneapi/setvars.sh
+fi
+
 if [[ ! -z "${ACCELERATE}" ]] && [ ${ACCELERATE}="True" ] && [ -x "$(command -v accelerate)" ]
 then
     echo "Accelerating launch.py..."
     exec accelerate launch --num_cpu_threads_per_process=6 launch.py "$@"
+elif [[ -z "${first_launch}" ]] && [ -x "$(command -v ipexrun)" ] && [[ "$@" == *"--use-ipex"* ]]
+then
+    echo "Ipexrun'ning launch.py..."
+    exec ipexrun launch.py "$@"
 else
     echo "Launching launch.py..."
     exec "${python_cmd}" launch.py "$@"