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", + ] }