tritonparse 0.3.2.dev20251210071601__py3-none-any.whl

tritonparse-0.3.2.dev20251210071601.dist-info/METADATA

@@ -0,0 +1,195 @@
+Metadata-Version: 2.4
+Name: tritonparse
+Version: 0.3.2.dev20251210071601
+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
+Project-URL: Homepage, https://github.com/meta-pytorch/tritonparse
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+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
+
+[![License: BSD-3](https://img.shields.io/badge/License-BSD--3-blue.svg)](https://opensource.org/licenses/BSD-3-Clause)
+[![GitHub Pages](https://img.shields.io/badge/GitHub%20Pages-Deploy-brightgreen)](https://meta-pytorch.org/tritonparse/)
+
+**A comprehensive visualization and analysis tool for Triton kernel compilation and launch** — helping developers analyze, debug, and understand Triton kernel compilation processes.
+
+🌐 **[Try it online →](https://meta-pytorch.org/tritonparse/?json_url=https://meta-pytorch.org/tritonparse/dedicated_log_triton_trace_findhao__mapped.ndjson.gz)**
+
+## ✨ Key Features
+
+### 🔍 Visualization & Analysis
+- **🚀 Launch Difference Analysis** - Detect and visualize kernel launch parameter variations
+- **📊 IR Code View** - Side-by-side IR viewing with synchronized highlighting and line mapping
+- **🔄 File Diff View** - Compare kernels across different trace files side-by-side
+- **📝 Multi-format IR Support** - View TTGIR, TTIR, LLIR, PTX, and AMDGCN
+- **🎯 Interactive Code Views** - Click-to-highlight corresponding lines across IR stages
+
+### 🔧 Reproducer & Debugging Tools
+- **🔄 Standalone Script Generation** - Extract any kernel into a self-contained Python script
+- **💾 Tensor Data Reconstruction** - Preserve actual tensor data or use statistical approximation
+- **🎯 Custom Templates** - Flexible reproducer templates for different workflows
+- **🐛 Bug Isolation** - Share reproducible test cases for debugging and collaboration
+
+### 📊 Structured Logging & Analysis
+- **📝 Compilation & Launch Tracing** - Capture detailed events with source mapping
+- **🔍 Stack Trace Integration** - Full Python stack traces for debugging
+- **📈 Metadata Extraction** - Comprehensive kernel statistics
+
+### 🛠️ Developer Tools
+- **🌐 Browser-based Interface** - No installation required, works in your browser
+- **🔒 Privacy-first** - All processing happens locally, no data uploaded
+
+## 🚀 Quick Start
+
+### 1. Installation
+
+**Four options to install:**
+```bash
+# install nightly version
+pip install -U --pre tritonparse
+# install stable version
+pip install tritonparse
+# install from source
+git clone https://github.com/meta-pytorch/tritonparse.git
+cd tritonparse
+pip install -e .
+# pip install the latest version from github
+pip install git+https://github.com/meta-pytorch/tritonparse.git
+```
+
+**Prerequisites:** Python ≥ 3.10, Triton ≥ 3.4.0, GPU required (NVIDIA/AMD)
+
+TritonParse relies on new features in Triton. If you're using nightly PyTorch, Triton is already included. Otherwise, install the latest Triton:
+```bash
+pip install triton
+```
+
+### 2. Generate Traces
+
+```python
+import tritonparse.structured_logging
+import tritonparse.utils
+
+# Initialize logging
+tritonparse.structured_logging.init("./logs/", enable_trace_launch=True)
+
+# Your Triton/PyTorch code here
+# ... your kernels ...
+
+# Parse and generate trace files
+tritonparse.utils.unified_parse("./logs/", out="./parsed_output")
+```
+
+<details>
+<summary>📝 Example output (click to expand)</summary>
+
+```bash
+================================================================================
+📁 TRITONPARSE PARSING RESULTS
+================================================================================
+📂 Parsed files directory: /scratch/findhao/tritonparse/tests/parsed_output
+📊 Total files generated: 2
+
+📄 Generated files:
+   1. 📝 dedicated_log_triton_trace_findhao__mapped.ndjson.gz (7.2KB)
+   2. 📝 log_file_list.json (181B)
+================================================================================
+✅ Parsing completed successfully!
+================================================================================
+```
+</details>
+
+### 3. Visualize Results
+
+**Visit [https://meta-pytorch.org/tritonparse/](https://meta-pytorch.org/tritonparse/?json_url=https://meta-pytorch.org/tritonparse/dedicated_log_triton_trace_findhao__mapped.ndjson.gz)** and open your local trace files (.ndjson.gz format).
+
+> **🔒 Privacy Note**: Your trace files are processed entirely in your browser - nothing is uploaded to any server!
+
+### 4. Generate Reproducers (Optional)
+
+Extract any kernel into a standalone, executable Python script for debugging or testing:
+
+```bash
+# Generate reproducer for the first launch event
+# (--line is 0-based: line 0 is compilation event, line 1 is first launch event)
+tritonparseoss reproduce ./parsed_output/trace.ndjson.gz --line 1 --out-dir repro_output
+
+# Run the generated reproducer
+cd repro_output/<kernel_name>/
+python repro_*.py
+```
+
+**Python API:**
+```python
+from tritonparse.reproducer.orchestrator import reproduce
+
+result = reproduce(
+    input_path="./parsed_output/trace.ndjson.gz",
+    line_index=0,           # 0-based index (first event is 0)
+    out_dir="repro_output"
+)
+```
+
+<details>
+<summary>🎯 Common Reproducer Use Cases (click to expand)</summary>
+
+- **🐛 Bug Isolation**: Extract a failing kernel into a minimal standalone script
+- **⚡ Performance Testing**: Benchmark specific kernels without running the full application
+- **🤝 Team Collaboration**: Share reproducible test cases with colleagues or in bug reports
+- **📊 Regression Testing**: Compare kernel behavior and performance across different versions
+- **🔍 Deep Debugging**: Modify and experiment with kernel parameters in isolation
+
+</details>
+
+## 📚 Complete Documentation
+
+| 📖 Guide | Description |
+|----------|-------------|
+| **[🏠 Wiki Home](https://github.com/meta-pytorch/tritonparse/wiki)** | Complete documentation and quick navigation |
+| **[📦 Installation](https://github.com/meta-pytorch/tritonparse/wiki/01.-Installation)** | Setup guide for all scenarios |
+| **[📋 Usage Guide](https://github.com/meta-pytorch/tritonparse/wiki/02.-Usage-Guide)** | Complete workflow, reproducer generation, and examples |
+| **[🌐 Web Interface](https://github.com/meta-pytorch/tritonparse/wiki/03.-Web-Interface-Guide)** | Master the visualization interface |
+| **[🔧 Developer Guide](https://github.com/meta-pytorch/tritonparse/wiki/04.-Developer-Guide)** | Contributing and architecture overview |
+| **[📝 Code Formatting](https://github.com/meta-pytorch/tritonparse/wiki/05.-Code-Formatting)** | Formatting standards and tools |
+| **[❓ FAQ](https://github.com/meta-pytorch/tritonparse/wiki/06.-FAQ)** | Quick answers and troubleshooting |
+
+## 📊 Understanding Triton Compilation
+
+TritonParse visualizes the complete Triton compilation pipeline:
+
+**Python Source** → **TTIR** → **TTGIR** → **LLIR** → **PTX/AMDGCN**
+
+Each stage can be inspected and compared to understand optimization transformations.
+
+## 🤝 Contributing
+
+We welcome contributions! Please see our **[Developer Guide](https://github.com/meta-pytorch/tritonparse/wiki/04.-Developer-Guide)** for:
+- Development setup and prerequisites
+- Code formatting standards (**[Formatting Guide](https://github.com/meta-pytorch/tritonparse/wiki/05.-Code-Formatting)**)
+- Pull request and code review process
+- Testing guidelines
+- Architecture overview
+
+## 📞 Support & Community
+
+- **🐛 Report Issues**: [GitHub Issues](https://github.com/meta-pytorch/tritonparse/issues)
+- **💬 Discussions**: [GitHub Discussions](https://github.com/meta-pytorch/tritonparse/discussions)
+- **📚 Documentation**: [TritonParse Wiki](https://github.com/meta-pytorch/tritonparse/wiki)
+
+## 📄 License
+
+This project is licensed under the BSD-3 License - see the [LICENSE](LICENSE) file for details.
+
+---
+
+**✨ Ready to get started?** Visit our **[Installation Guide](https://github.com/meta-pytorch/tritonparse/wiki/01.-Installation)** or try the **[online tool](https://meta-pytorch.org/tritonparse/)** directly!

tritonparse-0.3.2.dev20251210071601.dist-info/RECORD

@@ -0,0 +1,62 @@
+tritonparse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tritonparse/__main__.py,sha256=RXbkALBewcb1xlJBnsQl9IaBRUNln7U8NuRZKT8UdIk,117
+tritonparse/cli.py,sha256=xh5T4Mq0w93BG-A_87T_-FdLXv6ku4UDQU3RzmpxJgg,3577
+tritonparse/common.py,sha256=IXPeC68qmBG6ssDQNqBNFRGdUtJZ1V8jlEHXY5frEpw,13923
+tritonparse/context_manager.py,sha256=jKIVsgTjVQEbJXcXB99pB0OsHf1-w5sECwm5WN9LVKU,2242
+tritonparse/event_diff.py,sha256=USCjfjYr-7Ie-EfZgtCFMZMA1KRzFRDe7yDFy98zYI4,4962
+tritonparse/extract_source_mappings.py,sha256=Z6UxFj2cCE5NCWLQTYPKqUpLfbYhqP8xgCl5mvud9KI,1451
+tritonparse/ir_analysis.py,sha256=-BbBZwpRqpB4s_P044g7ZuwP_73g55P1uoi2n9oEC0Y,16634
+tritonparse/ir_parser.py,sha256=JQ7hsevmhFGmtZ3CoXi4utcomAycBQTT-KFjSva2K8U,15565
+tritonparse/mapper.py,sha256=QBCUMHM9pu3x3ahFp0wyXJbmv9TFGVPdkcLULok1E-k,4205
+tritonparse/shared_vars.py,sha256=RifXq55KisHgspYAmGcaCWY6ZHX8iejFHvwIewvcWZE,707
+tritonparse/source_type.py,sha256=nmYEQS8rfkIN9BhNhQbkmEvKnvS-3zAxRGLY4TaZdi8,1676
+tritonparse/sourcemap_utils.py,sha256=uI02n5Sgnlx7Nc15QAX5N6_tZZMips0PyJuo1n3eouY,2654
+tritonparse/structured_logging.py,sha256=mP3B23dk4vZAPrhBokRP_YXTj0Rjb8Rh8b-w_W5JSak,64053
+tritonparse/tp_logger.py,sha256=vXzY7hMDmVnRBGBhIjFZe3nHZzG5NKKPONGUszJhGgU,242
+tritonparse/trace_processor.py,sha256=gCPYiWJEaxaQMzwYmv8uJZav0gIJGxl12E6m8Cr7uvI,14367
+tritonparse/utils.py,sha256=Jnlptcd79llSDev-_1XyyOnv2izUqv0PEL74A8GF2tc,4565
+tritonparse/info/__init__.py,sha256=NsQpDxge4-ezfTeol1j-bMV8SpjlpW8hHNLI7G2T0FY,756
+tritonparse/info/cli.py,sha256=SVnEo7iX2Luu4EZR3dS3a7zxPvAoTezzwZnXpGahNUQ,4240
+tritonparse/info/kernel_query.py,sha256=4Zr2Y8p0z1IHnxLm72aj5_OHdCYkwAPsB_59oNWP77M,6124
+tritonparse/info/parse_helper.py,sha256=Kmk67PwdD3_TdyJoMSajdDnNJ3igH9usvRU0vY1aSj0,2323
+tritonparse/reproducer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tritonparse/reproducer/ast_analyzer.py,sha256=4S1cN8AXYBMh8FjkvZoTrOO_KIeJPQ_dhy13OQY1Vio,24131
+tritonparse/reproducer/cli.py,sha256=CRysJGDttucKTuxzUtgWvxVXU7UBD64KQHP_8l2CjvQ,2365
+tritonparse/reproducer/consolidated_result.py,sha256=c0CFx2YyIsO4bJZsLRSQ_y1QwpZl_PoiS4eH4_a79bk,1383
+tritonparse/reproducer/function_extractor.py,sha256=lBmIHPBPyBJd4UU4RafCEjlGzC2vgq8mOWda7Tt7oOA,6865
+tritonparse/reproducer/import_info.py,sha256=XoZSzIUSt54C5Ayi8bzqy9XepFmIHTEKO6q9zBUkvcg,840
+tritonparse/reproducer/import_parser.py,sha256=eHNAFkluefWcOnMId2DHidBzTex5vHEKtKjmLUPXNOA,5896
+tritonparse/reproducer/import_resolver.py,sha256=nlNknv0WBeGKNJJ-Cr0OROmJEObolx-v6cUkETK5kzQ,5223
+tritonparse/reproducer/multi_file_analyzer.py,sha256=STEh6UpSc4sKsvVbGO3y7RyNyhF_KlqRTjG-eo4APZU,30189
+tritonparse/reproducer/orchestrator.py,sha256=A8RAZKrxtvm5lBymI_ExgbT94wWvhEpvT2XmK__5TZw,4449
+tritonparse/reproducer/placeholder_replacer.py,sha256=VR4vsVVCBZnfJx5YMTwGd0UBQUeY6jN12nYolnjXmdk,13526
+tritonparse/reproducer/types.py,sha256=86wql3NaGgpkOzx0gDFb5qexNjKExzhL0uIwGU7grrw,564
+tritonparse/reproducer/utils.py,sha256=xkFIwAJOLxXLtKTOyVKiYR0ZKBDlHzeVijscz1dVETM,20446
+tritonparse/reproducer/ingestion/ndjson.py,sha256=7amSwpbtG-od1-pW18Nm9AiaFc3Etd0-UETXwiYCmgw,7443
+tritonparse/reproducer/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tritonparse/reproducer/templates/example.py,sha256=e1ux6LpE3L6qf2Oxf-F50spBRbtRbfbHcQJbhz2Xt04,882
+tritonparse/reproducer/templates/loader.py,sha256=x14KHXkovOIcXFKii3Jx4XjpEhXqUMqp575qAffi370,1975
+tritonparse/reproducer/templates/tritonbench.py,sha256=J6AkJtnWTA9XBInFIy7Dr-LD-mIviQcW4Wqq_x9an3A,2928
+tritonparse/reproducer/templates/utils.py,sha256=1Sczd4SQPmZPZLYTXgRc_brZq6LUDLFj5XNfgcljWYQ,1232
+tritonparse/reproducer/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tritonparse/reproducer/tests/test_import_parser.py,sha256=WGOC5xlc9GagsJvgoXh8rcIBEAM2URDTRT3igBq30Mc,6015
+tritonparse/reproducer/tests/test_import_resolver.py,sha256=rL0EWghqRtGP5fktCDnmTERVj1Lm7r2Al9pt46-sSKI,2983
+tritonparse/reproducer/tests/test_multi_file_analyzer.py,sha256=HhOoViZo49pPngongtSp2z70caFu_u9qqq7ymzZu2mA,4681
+tritonparse/reproducer/tests/artifacts/__init__.py,sha256=1GSd2uwSOqwtd-gOnJ9YD43LjaNANG9uwSGH3NWuTwo,144
+tritonparse/reproducer/tests/artifacts/triton_fused_kernel.py,sha256=F5lcCM9Yfl5VwVbTS86nJck-1hstYqtPEoUb1TCRBPo,1644
+tritonparse/reproducer/tests/artifacts/triton_preprocess.py,sha256=DXcH51cGInc4j7jttOGDsl2M6TWFB93dI_49Pn2OTZY,336
+tritonparse/reproducer/tests/artifacts/triton_utils.py,sha256=5_pZ-8AwHZW-MA5BHD6WzUodYCeXousdjHnj-v802Wg,211
+tritonparse/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tritonparse/tools/decompress_bin_ndjson.py,sha256=Gn5foDIlxBN5D5wmcdrEmwvxo3_wRlH8ih2U2Ys3RdM,4199
+tritonparse/tools/disasm.py,sha256=c4HmNNoPPeXPQBQkPVcMaHwDHbHNZNxuqXn4UIIs1Z0,2434
+tritonparse/tools/extract_irs.py,sha256=ByZMX0lOhpjmrpx23HVmASWZG5ow2bvqgpojElVFsmg,8052
+tritonparse/tools/format_fix.py,sha256=ISalg_N_L7Xktag3mLr-G9T6Opxv793s1WG6A9wUtXk,3922
+tritonparse/tools/load_tensor.py,sha256=7-LbpboKDNJFBLNhiKS3enoqRlVAb55OjPc70PwHXAw,2789
+tritonparse/tools/prettify_ndjson.py,sha256=3R2VNU__kgm5qZxkVa_fnZqBdEpkd4-GH12Lv7UG_P0,11651
+tritonparse/tools/readme.md,sha256=opFMUsu9igcn8COUSt_-JbNwEkbsmCQkdkuShjlqm_0,1112
+tritonparse-0.3.2.dev20251210071601.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
+tritonparse-0.3.2.dev20251210071601.dist-info/METADATA,sha256=-XkIeIG0Lqeh5m8PZY1dSaH3Rvs96eEmUlRheLVBysQ,8366
+tritonparse-0.3.2.dev20251210071601.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tritonparse-0.3.2.dev20251210071601.dist-info/entry_points.txt,sha256=wEXdaieDoRRCCdhEv2p_C68iytnaXU_2pwt5CqjfbWY,56
+tritonparse-0.3.2.dev20251210071601.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
+tritonparse-0.3.2.dev20251210071601.dist-info/RECORD,,

tritonparse-0.3.2.dev20251210071601.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

tritonparse-0.3.2.dev20251210071601.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+tritonparseoss = tritonparse.cli:main

tritonparse-0.3.2.dev20251210071601.dist-info/licenses/LICENSE

@@ -0,0 +1,29 @@
+BSD 3-Clause License
+
+Copyright (c) 2019, pytorch
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

tritonparse-0.3.2.dev20251210071601.dist-info/top_level.txt

@@ -0,0 +1 @@
+tritonparse