add 'pti_lib.py' script

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

Author: anmyachev

.github/workflows/build-test-reusable.yml

@@ -284,9 +284,11 @@ jobs:
         if: matrix.suite == 'rest'
         run: |
           ./scripts/install-pti.sh
+          PTI_LIBS_DIR=$(python ./scripts/pti_lib.py)
+          ls $PTI_LIBS_DIR
           LEVEL_ZERO_LIB=$(pwd)/.scripts_cache/level_zero_for_pti/level-zero-1.24.2/build/lib/
           ls $LEVEL_ZERO_LIB
-          export LD_LIBRARY_PATH=$LEVEL_ZERO_LIB:$LD_LIBRARY_PATH
+          export LD_LIBRARY_PATH=$PTI_LIBS_DIR:$LEVEL_ZERO_LIB:$LD_LIBRARY_PATH
           cd third_party/proton/test
           pytest test_api.py test_cmd.py test_lib.py test_profile.py test_viewer.py --device xpu -s -v
           cd ..

scripts/pti_lib.py

@@ -0,0 +1,22 @@
+"""Prints a lib directory for pti."""
+
+import importlib.metadata
+import pathlib
+
+
+def get_pti_lib_path() -> pathlib.Path:
+    """Returns library path for pti.
+
+    Raises:
+        importlib.metadata.PackageNotFoundError: if 'intel-pti' not installed.
+        AssertionError: if libpti_view.so not found.
+    """
+    files = importlib.metadata.files('intel-pti') or []
+    for f in files:
+        if f.name == 'libpti_view.so':
+            return pathlib.Path(f.locate()).parent.resolve()
+    raise AssertionError('libpti_view.so not found')
+
+
+if __name__ == '__main__':
+    print(get_pti_lib_path())