Fix xpu binary check env setup (#1972)

pytorch · Aug 22, 2024 · 18c9dd1 · 18c9dd1
1 parent 8d7f9a6
commit 18c9dd1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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
 ###############################################################################