Workflow to test Triton with pip dependencies (intel#2806)

Fixes intel#2795.

Author: pbchekin

.github/workflows/pip-test.yml

@@ -0,0 +1,76 @@
+name: Test with pip
+
+on:
+  workflow_dispatch:
+
+  # run workflow on changes to the driver, which handles the libraries logic
+  pull_request:
+    branches:
+      - main
+    paths:
+      - third_party/intel/backend/driver.py
+  push:
+    branches:
+      - main
+    paths:
+      - third_party/intel/backend/driver.py
+
+  # run workflow after building nightly wheels
+  workflow_run:
+    workflows:
+      - nightly-wheels.yml
+    types:
+      - completed
+
+permissions: read-all
+
+env:
+  PYTHON_VERSION: '3.9'
+  TRITON_TEST_CMD: "scripts/test-triton.sh --skip-pytorch-install"
+
+jobs:
+  tests:
+    name: Tests
+    if: ${{ github.event_name != 'workflow_run' || github.event.workflow_run.conclusion == 'success' }}
+    runs-on:
+      - rolling
+      - runner-0.0.20
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Install Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ env.PYTHON_VERSION }}
+
+      - name: Install wheels
+        uses: ./.github/actions/install-wheels
+        with:
+          gh_token: ${{ secrets.GITHUB_TOKEN }}
+          python_version: ${{ env.PYTHON_VERSION }}
+          # transformers package is required for the inductor (e2e) test
+          wheels_pattern: '{torch,transformers}-*.whl'
+
+      - name: Setup Triton
+        uses: ./.github/actions/setup-triton
+
+      - name: Install Triton runtime dependencies
+        run: |
+          pip install intel-sycl-rt intel-pti
+
+      - name: Run core tests
+        run: |
+          ${{ env.TRITON_TEST_CMD }} --core
+
+      - name: Run interpreter tests
+        run: |
+          ${{ env.TRITON_TEST_CMD }} --interpreter --skip-pip-install
+
+      - name: Run Tutorials
+        run: |
+          ${{ env.TRITON_TEST_CMD }} --tutorial --skip-pip-install
+
+      - name: Run inductor test
+        run: |
+          ${{ env.TRITON_TEST_CMD }} --inductor --skip-pip-install

scripts/sycl_rt_lib.py

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