From 18c9dd1deb9087894e527514e3463f7ec817807b Mon Sep 17 00:00:00 2001
From: "Wang, Chuanqi" <chuanqi.wang@intel.com>
Date: Thu, 22 Aug 2024 22:32:31 +0800
Subject: [PATCH] Fix xpu binary check env setup (#1972)

---
 check_binary.sh | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/check_binary.sh b/check_binary.sh
index db609979c..122e560d5 100755
--- a/check_binary.sh
+++ b/check_binary.sh
@@ -51,6 +51,15 @@ else
   install_root="$(dirname $(which python))/../lib/python${py_dot}/site-packages/torch/"
 fi
 
+###############################################################################
+# Setup XPU ENV
+###############################################################################
+if [[ "$DESIRED_CUDA" == 'xpu' ]]; then
+  # Refer https://www.intel.com/content/www/us/en/developer/articles/tool/pytorch-prerequisites-for-intel-gpu/2-5.html
+  source /opt/intel/oneapi/pytorch-gpu-dev-0.5/oneapi-vars.sh
+  source /opt/intel/oneapi/pti/latest/env/vars.sh
+fi
+
 ###############################################################################
 # Check GCC ABI
 ###############################################################################