tritonparse 0.2.4.dev20251006071528__tar.gz → 0.2.4.dev20251008071501__tar.gz

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251008071501}/.gitignore

@@ -70,4 +70,5 @@ env.bak/
 venv.bak/
 *.mdc
 repro_output/
+.github/copilot-instructions.md
 # end

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251008071501}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.2.4.dev20251006071528
+Version: 0.2.4.dev20251008071501
 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
@@ -27,13 +27,22 @@ Dynamic: license-file
 
 ## ✨ Key Features
 
-- **🚀 Launch Difference Analysis** - Automatically detect and visualize variations in kernel launch parameters, helping you pinpoint performance bottlenecks and debug launch configurations.
-- **🔍 Interactive Visualization** - Explore Triton kernels with detailed metadata and stack traces
-- **📊 Multi-format IR Support** - View TTGIR, TTIR, LLIR, PTX, and AMDGCN in one place
-- **🔄 Side-by-side Comparison** - Compare IR stages with synchronized highlighting
-- **📝 Structured Logging** - Capture detailed compilation and launch events with source mapping
-- **🌐 Ready-to-use Interface** - No installation required, works in your browser
-- **🔒 Privacy-first** - All processing happens locally in your browser, no data uploaded
+### 🔍 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
+
+### 📊 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
+- **🔧 Reproducer Generation** - Generate standalone Python scripts to reproduce kernels
+- **🌐 Browser-based Interface** - No installation required, works in your browser
+- **🔒 Privacy-first** - All processing happens locally, no data uploaded
 
 ## 🚀 Quick Start
 
@@ -41,22 +50,22 @@ Dynamic: license-file
 
 ```python
 import tritonparse.structured_logging
+import tritonparse.utils
 
-# Initialize logging with launch tracing enabled
+# Initialize logging
 tritonparse.structured_logging.init("./logs/", enable_trace_launch=True)
 
 # Your Triton/PyTorch code here
 # ... your kernels ...
 
 # Parse and generate trace files
-import tritonparse.utils
-tritonparse.utils.unified_parse("./logs/")
+tritonparse.utils.unified_parse("./logs/", out="./parsed_output")
 ```
-The example terminal output is:
-```bash
-tritonparse log file list: /tmp/tmp1gan7zky/log_file_list.json
-INFO:tritonparse:Copying parsed logs from /tmp/tmp1gan7zky to /scratch/findhao/tritonparse/tests/parsed_output
 
+<details>
+<summary>📝 Example output (click to expand)</summary>
+
+```bash
 ================================================================================
 📁 TRITONPARSE PARSING RESULTS
 ================================================================================
@@ -64,13 +73,13 @@ INFO:tritonparse:Copying parsed logs from /tmp/tmp1gan7zky to /scratch/findhao/t
 📊 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>
 
 ### 2. Visualize Results
 
@@ -106,18 +115,13 @@ pip install triton
 
 | 📖 Guide | Description |
 |----------|-------------|
-| **[🏠 Wiki Home](https://github.com/meta-pytorch/tritonparse/wiki)** | Complete documentation and navigation |
-| **[📦 Installation Guide](https://github.com/meta-pytorch/tritonparse/wiki/01.-Installation)** | Detailed setup for all scenarios |
-| **[📋 Usage Guide](https://github.com/meta-pytorch/tritonparse/wiki/02.-Usage-Guide)** | Complete workflow and examples |
-| **[🌐 Web Interface Guide](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 development setup |
-| **[❓ FAQ](https://github.com/meta-pytorch/tritonparse/wiki/06.-FAQ)** | Frequently asked questions |
-
-## 🛠️ Tech Stack
-
-- **Frontend**: React 19, TypeScript, Vite, Tailwind CSS, Monaco Editor
-- **Backend**: Python with Triton integration, structured logging
-- **Deployment**: GitHub Pages, automatic deployment
+| **[🏠 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, examples, and reproducer |
+| **[🌐 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
 
@@ -130,9 +134,10 @@ Each stage can be inspected and compared to understand optimization transformati
 ## 🤝 Contributing
 
 We welcome contributions! Please see our **[Developer Guide](https://github.com/meta-pytorch/tritonparse/wiki/04.-Developer-Guide)** for:
-- Development setup
-- Code formatting standards
-- Pull request process
+- 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

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251008071501}/README.md

@@ -9,13 +9,22 @@
 
 ## ✨ Key Features
 
-- **🚀 Launch Difference Analysis** - Automatically detect and visualize variations in kernel launch parameters, helping you pinpoint performance bottlenecks and debug launch configurations.
-- **🔍 Interactive Visualization** - Explore Triton kernels with detailed metadata and stack traces
-- **📊 Multi-format IR Support** - View TTGIR, TTIR, LLIR, PTX, and AMDGCN in one place
-- **🔄 Side-by-side Comparison** - Compare IR stages with synchronized highlighting
-- **📝 Structured Logging** - Capture detailed compilation and launch events with source mapping
-- **🌐 Ready-to-use Interface** - No installation required, works in your browser
-- **🔒 Privacy-first** - All processing happens locally in your browser, no data uploaded
+### 🔍 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
+
+### 📊 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
+- **🔧 Reproducer Generation** - Generate standalone Python scripts to reproduce kernels
+- **🌐 Browser-based Interface** - No installation required, works in your browser
+- **🔒 Privacy-first** - All processing happens locally, no data uploaded
 
 ## 🚀 Quick Start
 
@@ -23,22 +32,22 @@
 
 ```python
 import tritonparse.structured_logging
+import tritonparse.utils
 
-# Initialize logging with launch tracing enabled
+# Initialize logging
 tritonparse.structured_logging.init("./logs/", enable_trace_launch=True)
 
 # Your Triton/PyTorch code here
 # ... your kernels ...
 
 # Parse and generate trace files
-import tritonparse.utils
-tritonparse.utils.unified_parse("./logs/")
+tritonparse.utils.unified_parse("./logs/", out="./parsed_output")
 ```
-The example terminal output is:
-```bash
-tritonparse log file list: /tmp/tmp1gan7zky/log_file_list.json
-INFO:tritonparse:Copying parsed logs from /tmp/tmp1gan7zky to /scratch/findhao/tritonparse/tests/parsed_output
 
+<details>
+<summary>📝 Example output (click to expand)</summary>
+
+```bash
 ================================================================================
 📁 TRITONPARSE PARSING RESULTS
 ================================================================================
@@ -46,13 +55,13 @@ INFO:tritonparse:Copying parsed logs from /tmp/tmp1gan7zky to /scratch/findhao/t
 📊 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>
 
 ### 2. Visualize Results
 
@@ -88,18 +97,13 @@ pip install triton
 
 | 📖 Guide | Description |
 |----------|-------------|
-| **[🏠 Wiki Home](https://github.com/meta-pytorch/tritonparse/wiki)** | Complete documentation and navigation |
-| **[📦 Installation Guide](https://github.com/meta-pytorch/tritonparse/wiki/01.-Installation)** | Detailed setup for all scenarios |
-| **[📋 Usage Guide](https://github.com/meta-pytorch/tritonparse/wiki/02.-Usage-Guide)** | Complete workflow and examples |
-| **[🌐 Web Interface Guide](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 development setup |
-| **[❓ FAQ](https://github.com/meta-pytorch/tritonparse/wiki/06.-FAQ)** | Frequently asked questions |
-
-## 🛠️ Tech Stack
-
-- **Frontend**: React 19, TypeScript, Vite, Tailwind CSS, Monaco Editor
-- **Backend**: Python with Triton integration, structured logging
-- **Deployment**: GitHub Pages, automatic deployment
+| **[🏠 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, examples, and reproducer |
+| **[🌐 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
 
@@ -112,9 +116,10 @@ Each stage can be inspected and compared to understand optimization transformati
 ## 🤝 Contributing
 
 We welcome contributions! Please see our **[Developer Guide](https://github.com/meta-pytorch/tritonparse/wiki/04.-Developer-Guide)** for:
-- Development setup
-- Code formatting standards
-- Pull request process
+- 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

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251008071501}/pyproject.toml

@@ -48,4 +48,4 @@ local_scheme = "no-local-version"
 "Homepage" = "https://github.com/meta-pytorch/tritonparse"
 
 [project.scripts]
-tritonparse = "tritonparse.run:main"
+tritonparseoss = "tritonparse.cli:main"

tritonparse-0.2.4.dev20251008071501/run.py

@@ -0,0 +1,8 @@
+#!/usr/bin/env python3
+#  Copyright (c) Meta Platforms, Inc. and affiliates.
+
+from tritonparse.cli import main
+
+if __name__ == "__main__":
+    # Do not add code here, it won't be run. Add them to the function called below.
+    main()  # pragma: no cover

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251008071501}/tritonparse/__main__.py

@@ -1,4 +1,4 @@
-from .run import main
+from .cli import main
 
 
 if __name__ == "__main__":

tritonparse-0.2.4.dev20251006071528/run.py → tritonparse-0.2.4.dev20251008071501/tritonparse/cli.py

@@ -1,9 +1,9 @@
 #!/usr/bin/env python3
 #  Copyright (c) Meta Platforms, Inc. and affiliates.
-
 import argparse
 from importlib.metadata import PackageNotFoundError, version
 
+from .common import is_fbcode
 from .reproducer.cli import _add_reproducer_args
 from .reproducer.orchestrator import reproduce
 from .utils import _add_parse_args, unified_parse
@@ -19,15 +19,18 @@ def _get_package_version() -> str:
 def main():
     pkg_version = _get_package_version()
 
+    # Use different command name for fbcode vs OSS
+    prog_name = "tritonparse" if is_fbcode() else "tritonparseoss"
+
     parser = argparse.ArgumentParser(
-        prog="tritonparse",
+        prog=prog_name,
         description=(
             "TritonParse: parse structured logs and generate minimal reproducers"
         ),
         epilog=(
             "Examples:\n"
-            "  tritonparse parse /path/to/logs --out parsed_output\n"
-            "  tritonparse reproduce /path/to/trace.ndjson --line 1 --out-dir repro_output\n"
+            f"  {prog_name} parse /path/to/logs --out parsed_output\n"
+            f"  {prog_name} reproduce /path/to/trace.ndjson --line 2 --out-dir repro_output\n"
         ),
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )

{tritonparse-0.2.4.dev20251006071528 → tritonparse-0.2.4.dev20251008071501}/tritonparse/context_manager.py

@@ -17,6 +17,8 @@ class TritonParseManager:
         self,
         enable_trace_launch=False,
         split_inductor_compilations=True,
+        enable_tensor_blob_storage=False,
+        tensor_storage_quota=None,
         **parse_kwargs,
     ):
         """
@@ -25,17 +27,28 @@ class TritonParseManager:
         Args:
             enable_trace_launch: Whether to enable trace launch
             split_inductor_compilations: Whether to split inductor compilations in the output
+            enable_tensor_blob_storage: Whether to enable tensor blob storage
+            tensor_storage_quota: Storage quota in bytes for tensor blobs (default: 100GB)
             **parse_kwargs: Additional keyword arguments to pass to unified_parse
         """
         self.enable_trace_launch = enable_trace_launch
         self.split_inductor_compilations = split_inductor_compilations
+        self.enable_tensor_blob_storage = enable_tensor_blob_storage
+        self.tensor_storage_quota = tensor_storage_quota
         self.parse_kwargs = parse_kwargs
         self.dir_path = None
         self.output_link = None
 
     def __enter__(self):
         self.dir_path = createUniqueTempDirectory()
-        init(self.dir_path, enable_trace_launch=self.enable_trace_launch)
+        init_kwargs = {
+            "enable_trace_launch": self.enable_trace_launch,
+            "enable_tensor_blob_storage": self.enable_tensor_blob_storage,
+        }
+        if self.tensor_storage_quota is not None:
+            init_kwargs["tensor_storage_quota"] = self.tensor_storage_quota
+
+        init(self.dir_path, **init_kwargs)
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):