From fe499cf9a41c638b0fdb8acf13fea9f7446f3fc6 Mon Sep 17 00:00:00 2001
From: "Wang, Chuanqi" <chuanqi.wang@intel.com>
Date: Sat, 23 Nov 2024 05:37:19 +0800
Subject: [PATCH] [xpu] Add the torch xpu runtime dependencies pypi packages
 (#5965)

Add torch xpu runtime dependencies pypi packages into S3 manage
Works for https://github.com/pytorch/pytorch/issues/139722 and
https://github.com/pytorch/pytorch/issues/114850

---------

Co-authored-by: Huy Do <huydhn@gmail.com>
---
 s3_management/manage.py              | 11 +++++++++++
 s3_management/update_dependencies.py | 13 +++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/s3_management/manage.py b/s3_management/manage.py
index 79df63d659..71115ebf2a 100755
--- a/s3_management/manage.py
+++ b/s3_management/manage.py
@@ -102,6 +102,17 @@
     "importlib_metadata",
     "importlib_resources",
     "zipp",
+    # ---- torch xpu additional packages ----
+    "dpcpp-cpp-rt",
+    "intel-cmplr-lib-rt",
+    "intel-cmplr-lib-ur",
+    "intel-cmplr-lic-rt",
+    "intel-opencl-rt",
+    "intel-sycl-rt",
+    "intel-openmp",
+    "tcmlib",
+    "umf",
+    "intel-pti",
     # ----
     "Pillow",
     "certifi",
diff --git a/s3_management/update_dependencies.py b/s3_management/update_dependencies.py
index c9ee3697fc..3b253d9929 100644
--- a/s3_management/update_dependencies.py
+++ b/s3_management/update_dependencies.py
@@ -79,6 +79,19 @@
         "xxhash",
         "yarl",
     ]
+    "torch_xpu": [
+        "dpcpp-cpp-rt",
+        "intel-cmplr-lib-rt",
+        "intel-cmplr-lib-ur",
+        "intel-cmplr-lic-rt",
+        "intel-opencl-rt",
+        "intel-sycl-rt",
+        "intel-openmp",
+        "tcmlib",
+        "umf",
+        "intel-pti",
+        "tbb",
+    ]
 }