Add Parallel CI for Pip-Installed Triton and Update Docs (#48)

Summary:
Pre-built Triton has been upgraded to 3.4.0! So TritonParse can work out-of-box rather than compiling triton from source now!

This PR introduces parallel CI testing for Triton installed via pip and updates the documentation to recommend pip for standard installations, ensuring broader compatibility and easier setup for users.

## Summary of Changes

### CI/CD Workflow Enhancements

- **Added Parallel CI Job for Pip-Installed Triton**:
  - The main GitHub Actions workflow in `.github/workflows/test.yml` has been updated to include two parallel testing jobs:
    1.  `test-from-source`: The existing job, which tests against the latest Triton commit compiled from source.
    2.  `test-from-pip`: A new job that tests against the latest Triton version installed from PyPI.
  - This parallel setup ensures that `tritonparse` remains compatible with both source-built and pip-installed Triton environments.

- **New CI Script for Pip Installation**:
  - Created a new script, `.ci/install-triton-pip.sh`, to handle the installation of Triton from PyPI in the new `test-from-pip` CI job.

### Documentation Updates

- **Recommended Pip for Standard Installation**:
  - Updated the root `README.md` to recommend `pip install triton` as the standard installation method, removing the requirement to build from source.
  - Updated the wiki documentation, including `01.-Installation.md` and `04.-Developer-Guide.md`, to reflect that Triton should be installed via pip. Instructions for building from source have been replaced with the simpler pip command.

These changes improve the project by simplifying the installation process for users and expanding CI coverage to validate compatibility with official Triton releases from PyPI.

Pull Request resolved: #48

Reviewed By: adamomainz

Differential Revision: D79290083

Pulled By: FindHao

fbshipit-source-id: 9c7049b077f8c8d09812ac71f744efa0aa1deb4f

Author: FindHao

.ci/install-triton-pip.sh

@@ -0,0 +1,51 @@
+#!/bin/bash
+
+# Install Triton from pip
+# This script installs the latest version of Triton from PyPI.
+
+set -e
+
+echo "🚀 Installing Triton from pip..."
+START_TIME=$(date +%s)
+
+# Function to show elapsed time
+show_elapsed() {
+    CURRENT_TIME=$(date +%s)
+    ELAPSED=$((CURRENT_TIME - START_TIME))
+    echo "⏱️ Elapsed time: ${ELAPSED}s"
+}
+
+# Pre-flight checks
+echo "🔍 Running pre-flight checks..."
+
+# Ensure we're in the conda environment
+if [ -z "$CONDA_ENV" ]; then
+    echo "ERROR: CONDA_ENV is not set"
+    exit 1
+fi
+
+# Activate conda environment
+source /opt/miniconda3/etc/profile.d/conda.sh
+conda activate "$CONDA_ENV"
+
+# Uninstall existing triton to avoid conflicts
+echo "Uninstalling existing Triton versions..."
+pip uninstall -y pytorch-triton triton || true
+
+# Install Triton from pip
+echo "Installing the latest Triton from PyPI..."
+pip install triton
+
+show_elapsed
+
+# Verify Triton installation
+echo "Verifying Triton installation..."
+if python -c "import triton; print(f'Triton version: {triton.__version__}')" 2>/dev/null; then
+    python -c "import triton; print(f'Triton path: {triton.__file__}')"
+    echo "✅ Triton installation verified successfully"
+    show_elapsed
+    echo "🎉 Triton installation completed successfully!"
+else
+    echo "❌ ERROR: Failed to import Triton"
+    exit 1
+fi

.github/workflows/test.yml

@@ -56,7 +56,7 @@ jobs:
               run: |
                   make lint-check || (echo "❌ Linting failed. Please run 'make format' to fix formatting issues, then commit the changes." && echo "📖 For detailed formatting guide, see: https://github.com/pytorch-labs/tritonparse/wiki/05.-Code-Formatting" && exit 1)
 
-    test:
+    test-from-source:
         runs-on: 4-core-ubuntu-gpu-t4
         timeout-minutes: 120
         steps:
@@ -156,3 +156,58 @@ jobs:
                   COVERAGE: ${{ github.event.inputs.coverage || 'false' }}
               run: |
                   bash .ci/run-tests.sh
+
+    test-from-pip:
+        runs-on: 4-core-ubuntu-gpu-t4
+        timeout-minutes: 120
+        steps:
+            - uses: actions/checkout@v4
+
+            - name: Set up Python 3.11
+              uses: actions/setup-python@v4
+              with:
+                  python-version: "3.11"
+
+            - name: Get daily cache timestamp
+              id: daily-cache
+              run: |
+                  # Calculate date (e.g., 2024-01-15) for daily cache expiration
+                  DATE_STAMP=$(date +"%Y-%m-%d")
+                  echo "date=$DATE_STAMP" >> $GITHUB_OUTPUT
+                  echo "Using daily cache stamp: $DATE_STAMP"
+
+            - name: Cache pip dependencies
+              uses: actions/cache@v3
+              with:
+                  path: ~/.cache/pip
+                  key: ${{ runner.os }}-pip-3.11-${{ steps.daily-cache.outputs.date }}
+                  restore-keys: |
+                      ${{ runner.os }}-pip-3.11-
+
+            - name: Setup environment
+              env:
+                  CONDA_ENV: tritonparse-pip
+                  PYTHON_VERSION: "3.11"
+                  CUDA_VERSION: "12.8"
+              run: |
+                  bash .ci/setup.sh
+
+            - name: Install Triton from pip
+              env:
+                  CONDA_ENV: tritonparse-pip
+              run: |
+                  bash .ci/install-triton-pip.sh
+
+            - name: Install project dependencies
+              env:
+                  CONDA_ENV: tritonparse-pip
+              run: |
+                  bash .ci/install-project.sh
+
+            - name: Run tests
+              env:
+                  CONDA_ENV: tritonparse-pip
+                  TEST_TYPE: ${{ github.event.inputs.test-type || 'all' }}
+                  COVERAGE: ${{ github.event.inputs.coverage || 'false' }}
+              run: |
+                  bash .ci/run-tests.sh

README.md

@@ -69,9 +69,12 @@ cd tritonparse
 pip install -e .
 ```
 
-**Prerequisites:** Python ≥ 3.10, Triton > 3.3.1 ([install from source](https://github.com/triton-lang/triton)), GPU required (NVIDIA/AMD)
+**Prerequisites:** Python ≥ 3.10, Triton ≥ 3.4.0, GPU required (NVIDIA/AMD)
 
-TritonParse relies on new features in Triton > 3.3.1. Please install Triton from source for now.
+TritonParse relies on new features in Triton. Please install the latest version of Triton:
+```bash
+pip install triton
+```
 
 ## 📚 Complete Documentation
 

tritonparse/structured_logging.py

@@ -351,7 +351,11 @@ def extract_python_source_info(trace_data: Dict[str, Any], source):
         return
 
     # Get the original Python source code for the kernel
-    if isinstance(fn := source.fn, JITFunction) and hasattr(fn, "starting_line_number"):
+    if (
+        isinstance(fn := source.fn, JITFunction)
+        and hasattr(fn, "starting_line_number")
+        and hasattr(fn, "raw_src")
+    ):
         start_line_number = fn.starting_line_number
         source_lines = fn.raw_src
     else: