tritonparse 0.2.1.dev20250913071432__tar.gz → 0.2.1.dev20250914071450__tar.gz

{tritonparse-0.2.1.dev20250913071432 → tritonparse-0.2.1.dev20250914071450}/.github/workflows/test.yml

@@ -192,12 +192,6 @@ jobs:
               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

{tritonparse-0.2.1.dev20250913071432/tritonparse.egg-info → tritonparse-0.2.1.dev20250914071450}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.2.1.dev20250913071432
+Version: 0.2.1.dev20250914071450
 Summary: TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer Generator for Triton Kernels
 Author-email: Yueming Hao <yhao@meta.com>
 License-Expression: BSD-3-Clause
@@ -8,7 +8,10 @@ Project-URL: Homepage, https://github.com/meta-pytorch/tritonparse
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: triton>3.3.1
+Provides-Extra: triton
+Requires-Dist: triton>3.3.1; extra == "triton"
+Provides-Extra: pytorch-triton
+Requires-Dist: pytorch-triton>=3.4.0; extra == "pytorch-triton"
 Provides-Extra: test
 Requires-Dist: coverage>=7.0.0; extra == "test"
 Dynamic: license-file

{tritonparse-0.2.1.dev20250913071432 → tritonparse-0.2.1.dev20250914071450}/pyproject.toml

@@ -5,9 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "tritonparse"
 dynamic = ["version"]
-dependencies = [
-    "triton>3.3.1",
-]
+dependencies = []
 requires-python = ">=3.10"
 description = "TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer Generator for Triton Kernels"
 readme = "README.md"
@@ -17,8 +15,13 @@ authors = [
 license = "BSD-3-Clause"
 license-files = ["LICENSE*"]
 
-
 [project.optional-dependencies]
+triton = [
+    "triton>3.3.1",
+]
+pytorch-triton = [
+    "pytorch-triton>=3.4.0",
+]
 test = [
     "coverage>=7.0.0",
 ]

{tritonparse-0.2.1.dev20250913071432 → tritonparse-0.2.1.dev20250914071450}/tritonparse/common.py

@@ -357,15 +357,13 @@ def parse_logs(
                     file_mapping[rank_key]["mapped_file"] = mapped_file
 
     # Clean up the file mapping - remove None mapped_files and ensure no duplicates
-    for rank_key in file_mapping:
+    for rank_key, rank_data in file_mapping.items():
         if rank_key != "tritonparse_url_prefix":
             # Remove duplicates from regular_files
-            file_mapping[rank_key]["regular_files"] = list(
-                set(file_mapping[rank_key]["regular_files"])
-            )
+            rank_data["regular_files"] = list(set(rank_data["regular_files"]))
             # Remove mapped_file if None
-            if file_mapping[rank_key]["mapped_file"] is None:
-                del file_mapping[rank_key]["mapped_file"]
+            if rank_data["mapped_file"] is None:
+                del rank_data["mapped_file"]
     # Save file mapping to parsed_log_dir
     log_file_list_path = os.path.join(parsed_log_dir, "log_file_list.json")
     with open(log_file_list_path, "w") as f:

{tritonparse-0.2.1.dev20250913071432 → tritonparse-0.2.1.dev20250914071450/tritonparse.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.2.1.dev20250913071432
+Version: 0.2.1.dev20250914071450
 Summary: TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer Generator for Triton Kernels
 Author-email: Yueming Hao <yhao@meta.com>
 License-Expression: BSD-3-Clause
@@ -8,7 +8,10 @@ Project-URL: Homepage, https://github.com/meta-pytorch/tritonparse
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: triton>3.3.1
+Provides-Extra: triton
+Requires-Dist: triton>3.3.1; extra == "triton"
+Provides-Extra: pytorch-triton
+Requires-Dist: pytorch-triton>=3.4.0; extra == "pytorch-triton"
 Provides-Extra: test
 Requires-Dist: coverage>=7.0.0; extra == "test"
 Dynamic: license-file

{tritonparse-0.2.1.dev20250913071432 → tritonparse-0.2.1.dev20250914071450}/tritonparse.egg-info/SOURCES.txt

@@ -10,7 +10,6 @@ pyproject.toml
 run.py
 .ci/README.md
 .ci/install-project.sh
-.ci/install-triton-pip.sh
 .ci/install-triton.sh
 .ci/run-tests.sh
 .ci/setup.sh

tritonparse-0.2.1.dev20250914071450/tritonparse.egg-info/requires.txt

@@ -0,0 +1,9 @@
+
+[pytorch-triton]
+pytorch-triton>=3.4.0
+
+[test]
+coverage>=7.0.0
+
+[triton]
+triton>3.3.1

tritonparse-0.2.1.dev20250913071432/.ci/install-triton-pip.sh

@@ -1,67 +0,0 @@
-#!/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..."
-
-set +e  # Temporarily disable exit on error
-IMPORT_OUTPUT=$(python -c "import triton; print(f'Triton version: {triton.__version__}')" 2>&1)
-IMPORT_EXITCODE=$?
-set -e  # Re-enable exit on error
-
-echo "Import exit code: $IMPORT_EXITCODE"
-echo "Import output: $IMPORT_OUTPUT"
-
-if [ $IMPORT_EXITCODE -eq 0 ]; then
-    echo "$IMPORT_OUTPUT"
-    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"
-    echo "Import error details:"
-    echo "$IMPORT_OUTPUT"
-    echo ""
-    echo "Additional diagnostic information:"
-    echo "Installed packages containing 'triton':"
-    pip list | grep -i triton || echo "No triton packages found"
-    exit 1
-fi

tritonparse-0.2.1.dev20250913071432/tritonparse.egg-info/requires.txt

@@ -1,4 +0,0 @@
-triton>3.3.1
-
-[test]
-coverage>=7.0.0