[xpu] Add the torch xpu runtime dependencies pypi packages (#5965)

Add torch xpu runtime dependencies pypi packages into S3 manage Works for pytorch/pytorch#139722 and pytorch/pytorch#114850 --------- Co-authored-by: Huy Do <[email protected]>
pytorch · Nov 22, 2024 · fe499cf · fe499cf
1 parent fbf9fd1
commit fe499cf
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git 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
@@ -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",
+    ]
 }