Adapt code to use uv as a package manager (#5521)

Closes #4465

`Test with pip` is expected to fail until
pytorch/pytorch#166829 is merged.

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

scripts/pti_lib.py

@@ -13,7 +13,7 @@ def get_pti_lib_path() -> pathlib.Path:
     """
     files = importlib.metadata.files('intel-pti') or []
     for f in files:
-        if f.name in ('libpti_view.so', 'pti_view.lib'):
+        if any(map(lambda el: el in f.name, ('libpti_view.so', 'pti_view.lib'))):  # pylint: disable=W0640
             return pathlib.Path(f.locate()).parent.resolve()
     raise AssertionError('libpti_view.so not found')
 

scripts/sycl_rt_lib.py

@@ -13,7 +13,7 @@ def get_sycl_rt_lib_path() -> pathlib.Path:
     """
     files = importlib.metadata.files('intel-sycl-rt') or []
     for f in files:
-        if f.name == 'libsycl.so':
+        if 'libsycl.so' in f.name:
             return pathlib.Path(f.locate()).parent.resolve()
     raise AssertionError('libsycl.so not found')
 

scripts/torch_cmake.py

@@ -13,7 +13,7 @@ def get_torch_cmake_path() -> pathlib.Path:
     """
     files = importlib.metadata.files('torch') or []
     for f in files:
-        if f.name == 'TorchConfig.cmake':
+        if 'TorchConfig.cmake' in f.name:
             return pathlib.Path(f.locate()).parent.resolve()
     raise AssertionError('TorchConfig.cmake not found')
 

third_party/intel/backend/driver.py

@@ -69,9 +69,9 @@ def find_sycl(include_dir: list[str]) -> tuple[list[str], list[str]]:
     for f in importlib.metadata.files("intel-sycl-rt"):
         # sycl/sycl.hpp and sycl/CL/sycl.hpp results in both folders
         # being add: include and include/sycl.
-        if f.name == "sycl.hpp":
+        if "sycl.hpp" in f.name:
             include_dir += [str(f.locate().parent.parent.resolve())]
-        if f.name in ["libsycl.so", "sycl8.dll", "sycl8.lib"]:
+        if any(map(lambda el: el in f.name, ("libsycl.so", "sycl8.dll", "sycl8.lib"))):
             sycl_dir = str(f.locate().parent.resolve())
             # should we handle `_` somehow?
             if os.name == "nt":

third_party/proton/proton/profile.py

@@ -27,7 +27,7 @@ def _select_backend() -> str:
         try:
             if (files := importlib.metadata.files('intel-pti')) is not None:
                 for f in files:
-                    if f.name == 'libpti_view.so':
+                    if 'libpti_view.so' in f.name:
                         os.environ["TRITON_XPUPTI_LIB_PATH"] = str(pathlib.Path(f.locate()).parent.resolve())
                         break
         except importlib.metadata.PackageNotFoundError:

utils/SPIRVRunner/README.md

@@ -18,7 +18,7 @@ Alternatively, you can find `TorchConfig.cmake` with the following Python script
 import importlib.metadata
 
 for f in importlib.metadata.files('torch'):
-  if f.name == 'TorchConfig.cmake':
+  if 'TorchConfig.cmake' in f.name:
     print(f.locate().resolve())
 ```