tritonparse 0.3.1.dev20251026071454__tar.gz → 0.3.1.dev20251028071524__tar.gz

{tritonparse-0.3.1.dev20251026071454/tritonparse.egg-info → tritonparse-0.3.1.dev20251028071524}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.3.1.dev20251026071454
+Version: 0.3.1.dev20251028071524
 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
@@ -134,7 +134,7 @@ from tritonparse.reproducer.orchestrator import reproduce
 
 result = reproduce(
     input_path="./parsed_output/trace.ndjson.gz",
-    line_index=1,           # Which launch event (1-based)
+    line_index=0,           # 0-based index (first event is 0)
     out_dir="repro_output"
 )
 ```

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/README.md

@@ -116,7 +116,7 @@ from tritonparse.reproducer.orchestrator import reproduce
 
 result = reproduce(
     input_path="./parsed_output/trace.ndjson.gz",
-    line_index=1,           # Which launch event (1-based)
+    line_index=0,           # 0-based index (first event is 0)
     out_dir="repro_output"
 )
 ```

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/tests/test_tritonparse.py

@@ -187,6 +187,70 @@ class TestTritonparseCPU(unittest.TestCase):
 
         assert convert(nested_structure) == expected_nested
 
+    def test_loc_alias_parsing(self):
+        """Test parsing of location aliases in TTIR/TTGIR"""
+        from tritonparse.ir_parser import extract_loc_definitions
+
+        # Test case 1: Bare #loc reference (no number)
+        ir_with_bare_loc = """
+module {
+  #loc = loc("/tmp/test.py":10:5)
+  #loc13 = loc("x_ptr"(#loc))
+  func @kernel(%arg0: !tt.ptr<f32> loc(#loc13)) {
+    return loc(#loc)
+  }
+}
+"""
+        locs = extract_loc_definitions(ir_with_bare_loc)
+        # Main #loc should be stored with "" key
+        assert "" in locs, "Main #loc not found"
+        assert locs[""]["file"] == "/tmp/test.py"
+        assert locs[""]["line"] == 10
+        # Alias #loc13 should resolve to same location
+        assert "13" in locs, "#loc13 not found"
+        assert locs["13"]["file"] == "/tmp/test.py"
+        assert locs["13"]["line"] == 10
+        assert locs["13"]["alias_name"] == "x_ptr"
+        assert locs["13"]["alias_of"] == ""
+
+        # Test case 2: Named alias with numbered reference
+        ir_with_numbered_alias = """
+#loc = loc("/tmp/test.py":5:0)
+#loc2 = loc("/tmp/test.py":20:28)
+#loc16 = loc("pid"(#loc2))
+%0 = tt.get_program_id x : i32 loc(#loc16)
+"""
+        locs = extract_loc_definitions(ir_with_numbered_alias)
+        assert "2" in locs
+        assert locs["2"]["line"] == 20
+        assert "16" in locs
+        assert locs["16"]["file"] == "/tmp/test.py"
+        assert locs["16"]["line"] == 20
+        assert locs["16"]["alias_name"] == "pid"
+        assert locs["16"]["alias_of"] == "2"
+
+        # Test case 3: Simple alias (no name)
+        ir_with_simple_alias = """
+#loc = loc("/tmp/test.py":1:1)
+#loc1 = loc("/tmp/test.py":15:10)
+#loc20 = loc(#loc1)
+%1 = arith.constant 0 : i32 loc(#loc20)
+"""
+        locs = extract_loc_definitions(ir_with_simple_alias)
+        assert "1" in locs
+        assert "20" in locs
+        assert locs["20"]["file"] == "/tmp/test.py"
+        assert locs["20"]["line"] == 15
+        assert locs["20"]["alias_of"] == "1"
+        assert "alias_name" not in locs["20"]
+
+        # Test case 4: Definition line tracking
+        assert "def_line" in locs[""]
+        assert "def_line" in locs["1"]
+        assert "def_line" in locs["20"]
+
+        print("✓ All loc alias parsing tests passed")
+
 
 class TestTritonparseCUDA(unittest.TestCase):
     """CUDA tests (require GPU)"""

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/tritonparse/cli.py

@@ -70,7 +70,7 @@ def main():
     elif args.func == "reproduce":
         reproduce(
             input_path=args.input,
-            line_index=args.line,
+            line_index=args.line - 1,  # Convert 1-based line number to 0-based index
             out_dir=args.out_dir,
             template=args.template,
             kernel_import=args.kernel_import,

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/tritonparse/ir_analysis.py

@@ -42,7 +42,7 @@ def process_amd_gcn_bufferops(
 ) -> dict[str, int]:
     ir_content = load_ir_contents(key, file_content, file_path)
     # TODO: Add atomics
-    io_keys = ["global_load_", "global_store_", "buffer_load_", "buffer_store_"]
+    io_keys = ["global_load", "global_store", "buffer_load", "buffer_store"]
     return process_amd_bufferop(ir_content, io_keys)
 
 
@@ -66,9 +66,12 @@ def _generate_ir_analysis(entry: str):
         gcn_bufferops_info = process_amd_gcn_bufferops(
             amdgcn_key, file_content, file_path
         )
+        io_counts = {}
         # NDJSON format requires a newline at the end of each line
         if ttgir_bufferops_info:
-            ir_analysis["amd_ttgir_bufferops_count"] = ttgir_bufferops_info
+            io_counts["amd_ttgir_bufferops_count"] = ttgir_bufferops_info
         if gcn_bufferops_info:
-            ir_analysis["amd_gcn_bufferops_count"] = gcn_bufferops_info
-    return {"ir_analysis": ir_analysis}
+            io_counts["amd_gcn_bufferops_count"] = gcn_bufferops_info
+        if io_counts:
+            ir_analysis["io_counts"] = io_counts
+    return ir_analysis

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/tritonparse/ir_parser.py

@@ -10,7 +10,8 @@ logger = logging.getLogger("SourceMapping")
 
 # the definition of the #loc directive. they are in the bottom of the IR files
 # Example:#loc2 = loc("/tmp/torchinductor_yhao/yp/abcdef.py":20:28)
-LOC_PATTERN = re.compile(r'#loc(\d*) = loc\("([^"]+)":(\d+):(\d+)\)')
+# Note: This should only match numbered locs like #loc1, #loc2, not bare #loc
+LOC_PATTERN = re.compile(r'#loc(\d+) = loc\("([^"]+)":(\d+):(\d+)\)')
 
 # the reference to the #loc directive. they are in the end of lines of the IR files
 # Example: loc(#loc2)
@@ -33,6 +34,17 @@ AMDGCN_LOC_PATTERN = re.compile(
 )
 
 
+# alias loc definitions in TTGIR/TTIR
+# Example: #loc16 = loc("pid"(#loc2))
+# Example: #loc13 = loc("x_ptr"(#loc)) - bare #loc without number
+ALIAS_WITH_NAME_PATTERN = re.compile(
+    r'#loc(\d+)\s*=\s*loc\("([^"]+)"\s*\(\s*#loc(\d*)\s*\)\s*\)'
+)
+
+# Example: #loc20 = loc(#loc16)
+ALIAS_SIMPLE_PATTERN = re.compile(r"#loc(\d+)\s*=\s*loc\(\s*#loc(\d*)\s*\)")
+
+
 def extract_loc_definitions(ir_content: str) -> Dict[str, Dict[str, Any]]:
     """
     Extracts location definitions from the given IR content.
@@ -50,9 +62,10 @@ def extract_loc_definitions(ir_content: str) -> Dict[str, Dict[str, Any]]:
     """
     locations = {}
     # The first #loc directive is a special case. It locates at the top of the IR files
+    # Store it with empty string "" as key to avoid conflict with #loc1
     main_match = re.search(r'#loc = loc\("([^"]+)":(\d+):(\d+)\)', ir_content)
     if main_match:
-        locations["1"] = {
+        locations[""] = {
             "file": main_match.group(1),
             "line": int(main_match.group(2)),
             "column": int(main_match.group(3)),
@@ -61,6 +74,84 @@ def extract_loc_definitions(ir_content: str) -> Dict[str, Dict[str, Any]]:
     for loc_id, filename, line, col in LOC_PATTERN.findall(ir_content):
         key = loc_id
         locations[key] = {"file": filename, "line": int(line), "column": int(col)}
+
+    # Handle alias-style loc definitions that reference another #loc
+    # Build alias map first: alias_id -> target_id
+    alias_map: Dict[str, str] = {}
+    for m in ALIAS_WITH_NAME_PATTERN.finditer(ir_content):
+        alias_id, _name, target_id = m.groups()
+        # Empty target_id means bare #loc, map to "" (main loc key)
+        alias_map[alias_id] = target_id or ""
+    for m in ALIAS_SIMPLE_PATTERN.finditer(ir_content):
+        alias_id, target_id = m.groups()
+        # Empty target_id means bare #loc, map to "" (main loc key)
+        alias_map[alias_id] = target_id or ""
+
+    # Build definition line map and alias name map by scanning lines
+    def_line_map: Dict[str, int] = {}
+    alias_name_map: Dict[str, str] = {}
+    main_loc_line: int = 0
+    for i, line in enumerate(ir_content.split("\n"), start=1):
+        if m := ALIAS_WITH_NAME_PATTERN.search(line):
+            alias_id, name, target_id = m.groups()
+            def_line_map[alias_id] = i
+            alias_name_map[alias_id] = name
+            # ensure alias map is populated even if only found in line scan
+            # Empty target_id means bare #loc, map to "" (main loc key)
+            alias_map.setdefault(alias_id, target_id or "")
+        elif m := ALIAS_SIMPLE_PATTERN.search(line):
+            alias_id, target_id = m.groups()
+            def_line_map[alias_id] = i
+            # Empty target_id means bare #loc, map to "" (main loc key)
+            alias_map.setdefault(alias_id, target_id or "")
+        if m2 := LOC_PATTERN.search(line):
+            base_id, _fn, _ln, _col = m2.groups()
+            def_line_map[base_id] = i
+        if re.search(r'#loc\s*=\s*loc\("[^"]+":\d+:\d+\)', line):
+            # main #loc = loc("file":line:col) without id
+            main_loc_line = main_loc_line or i
+
+    # Resolve aliases to base locations (file/line/column)
+    resolving_stack = set()
+
+    def resolve_alias(current_id: str) -> Dict[str, Any]:
+        # Already a concrete location
+        if current_id in locations:
+            return locations[current_id]
+        # Detect cycles
+        if current_id in resolving_stack:
+            return {}
+        resolving_stack.add(current_id)
+        parent_id = alias_map.get(current_id)
+        result: Dict[str, Any] = {}
+        if parent_id is not None:
+            base = resolve_alias(parent_id)
+            if base:
+                # copy to avoid sharing the same dict by reference
+                result = {
+                    "file": base.get("file"),
+                    "line": base.get("line"),
+                    "column": base.get("column"),
+                }
+                locations[current_id] = result
+        resolving_stack.remove(current_id)
+        return result
+
+    # Resolve aliases and attach alias metadata
+    for alias_id, target_id in alias_map.items():
+        if alias_id not in locations:
+            resolve_alias(alias_id)
+        if alias_id in locations:
+            locations[alias_id]["alias_of"] = target_id
+            if alias_id in alias_name_map:
+                locations[alias_id]["alias_name"] = alias_name_map[alias_id]
+
+    # Attach definition line metadata
+    for k, v in def_line_map.items():
+        if k in locations:
+            locations[k]["def_line"] = v
+    if main_loc_line and "" in locations:
+        locations[""]["def_line"] = main_loc_line
     return locations
 
 

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/tritonparse/reproducer/ingestion/ndjson.py

@@ -40,7 +40,7 @@ def get_launch_and_compilation_events(
 
     Args:
         events: List of parsed event dictionaries.
-        line_index: Index of the launch event to process.
+        line_index: 0-based index of the launch event to process.
 
     Returns:
         Tuple of (launch_event, compilation_event).
@@ -179,7 +179,7 @@ def build_context_bundle(
 
     Args:
         events: List of parsed event dictionaries.
-        line_index: Index of the launch event to process.
+        line_index: 0-based index of the launch event to process.
 
     Returns:
         ContextBundle containing all information needed to reproduce the kernel launch.

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/tritonparse/reproducer/orchestrator.py

@@ -29,7 +29,7 @@ def reproduce(
 
     Args:
         input_path: Path to the NDJSON trace file.
-        line_index: Line index of the launch event to reproduce.
+        line_index: 0-based index of the launch event to reproduce in the events list.
         out_dir: Output directory for reproducer files.
         template: Template name to use for the reproducer.
         replacer: Optional custom PlaceholderReplacer instance. If None, uses DefaultPlaceholderReplacer.

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/tritonparse/trace_processor.py

@@ -71,12 +71,38 @@ def generate_source_mappings(
             }
         elif loc_id in loc_defs:
             info = loc_defs[loc_id]
-            mappings[str(ln)] = {
+            entry = {
                 "file": info["file"],
                 "line": info["line"],
                 "column": info["column"],
                 f"{ir_type}_line": ln,
             }
+            # Propagate alias metadata if present
+            if "alias_name" in info:
+                entry["alias_name"] = info["alias_name"]
+            if "alias_of" in info:
+                entry["loc_id"] = loc_id
+            mappings[str(ln)] = entry
+
+    # Add separate entries for loc definition lines
+    for loc_id, info in loc_defs.items():
+        if "def_line" not in info:
+            continue
+        def_ln = info["def_line"]
+        # Only create mapping if this line doesn't already have one
+        if str(def_ln) not in mappings:
+            entry = {
+                "file": info["file"],
+                "line": info["line"],
+                "column": info["column"],
+                f"{ir_type}_line": def_ln,
+                "kind": "loc_def",
+            }
+            if "alias_name" in info:
+                entry["alias_name"] = info["alias_name"]
+            if "alias_of" in info:
+                entry["loc_id"] = loc_id
+            mappings[str(def_ln)] = entry
 
     return mappings
 
@@ -301,8 +327,13 @@ def parse_single_file(
             )
 
         if compilation_event:
-            ir_analysis_event = _generate_ir_analysis(compilation_event)
-            if ir_analysis_event:
+            ir_analysis = _generate_ir_analysis(compilation_event)
+            if ir_analysis:
+                ir_analysis_event = {
+                    "event_type": "ir_analysis",
+                    "hash": _kernel_hash,
+                    "ir_analysis": ir_analysis,
+                }
                 all_output_lines[output_file].append(
                     json.dumps(ir_analysis_event, separators=(",", ":")) + "\n"
                 )

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524/tritonparse.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.3.1.dev20251026071454
+Version: 0.3.1.dev20251028071524
 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
@@ -134,7 +134,7 @@ from tritonparse.reproducer.orchestrator import reproduce
 
 result = reproduce(
     input_path="./parsed_output/trace.ndjson.gz",
-    line_index=1,           # Which launch event (1-based)
+    line_index=0,           # 0-based index (first event is 0)
     out_dir="repro_output"
 )
 ```

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/tritonparse.egg-info/SOURCES.txt

@@ -107,6 +107,7 @@ website/src/components/WelcomeScreen.tsx
 website/src/context/FileDiffSession.tsx
 website/src/pages/CodeView.tsx
 website/src/pages/FileDiffView.tsx
+website/src/pages/IRAnalysis.tsx
 website/src/pages/KernelOverview.tsx
 website/src/utils/dataLoader.ts
 website/src/utils/fbDetection.ts

tritonparse-0.3.1.dev20251028071524/website/index.html

@@ -0,0 +1,24 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <link rel="icon" type="image/svg+xml" href="/logo.svg" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>TritonParse</title>
+  </head>
+  <body>
+    <div id="root"></div>
+    <script type="module" src="/src/main.tsx"></script>
+
+    <!-- Static footer for SEO and tools that need to detect content in HTML source -->
+    <footer hidden>
+      <div>
+        Copyright © 2025 Meta Platforms, Inc
+        <span>|</span>
+        <a href="https://opensource.fb.com/legal/terms">Terms of Use</a>
+        <span>|</span>
+        <a href="https://opensource.fb.com/legal/privacy">Privacy Policy</a>
+      </div>
+    </footer>
+  </body>
+</html>

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/website/src/App.tsx

@@ -12,6 +12,7 @@ import CodeView from "./pages/CodeView";
 import FileDiffView from "./pages/FileDiffView";
 import SingleCodeViewer from "./components/SingleCodeViewer";
 import KernelOverview from "./pages/KernelOverview";
+import IRAnalysis from "./pages/IRAnalysis";
 import DataSourceSelector from "./components/DataSourceSelector";
 import WelcomeScreen from "./components/WelcomeScreen";
 import ExternalLink from "./components/ExternalLink";
@@ -409,7 +410,7 @@ function App() {
         </div>
       );
     } else {
-      // Show either overview, IR code, or file diff based on active tab
+      // Show either overview, IR code, IR analysis, or file diff based on active tab
       if (activeTab === "overview") {
         return (
           <KernelOverview
@@ -420,6 +421,14 @@ function App() {
           />
         );
       }
+      if (activeTab === "ir_analysis") {
+        return (
+          <IRAnalysis
+            kernels={kernels}
+            selectedKernel={selectedKernel}
+          />
+        );
+      }
       if (activeTab === "comparison") {
         return (
           <CodeView
@@ -550,7 +559,6 @@ function App() {
                   </button>
                 </>
               )}
-
               <button
                 className={`px-3 py-2 text-sm font-medium rounded-md ${activeTab === "file_diff" ? "bg-blue-700 text-white shadow-md" : "bg-blue-100 text-blue-700 hover:bg-blue-200"
                   }`}
@@ -561,6 +569,24 @@ function App() {
               >
                 File Diff
               </button>
+              {dataLoaded && kernels.length > 0 && (
+              <button
+                    className={`px-3 py-2 text-sm font-medium rounded-md ${activeTab === "ir_analysis" ? "bg-blue-700 text-white shadow-md" : "bg-blue-100 text-blue-700 hover:bg-blue-200"
+                      }`}
+                    onClick={() => {
+                      if (sess.preview?.active) sess.clearPreview();
+                      setActiveTab("ir_analysis");
+
+                      if (loadedUrl) {
+                        const newUrl = new URL(window.location.href);
+                        newUrl.searchParams.set("view", "ir_analysis");
+                        window.history.replaceState({}, "", newUrl.toString());
+                      }
+                    }}
+                  >
+                    IR Analysis (Beta)
+                  </button>
+              )}
             </div>
           </div>
         </div>

tritonparse-0.3.1.dev20251028071524/website/src/pages/IRAnalysis.tsx

@@ -0,0 +1,98 @@
+import React from "react";
+import { ProcessedKernel } from "../utils/dataLoader";
+
+interface IRAnalysisProps {
+  kernels: ProcessedKernel[];
+  selectedKernel: number;
+}
+
+const IRAnalysis: React.FC<IRAnalysisProps> = ({ kernels, selectedKernel }) => {
+  if (kernels.length === 0) {
+    return (
+      <div className="flex items-center justify-center h-screen">
+        <div className="text-gray-800">No kernel data available</div>
+      </div>
+    );
+  }
+
+  const kernel = kernels[selectedKernel];
+  if (kernel.ir_analysis === null) {
+    return (
+      <div className="flex items-center justify-center h-screen">
+        <div className="text-gray-800">No IR Analysis available</div>
+      </div>
+    );
+  }
+
+  const io_counts = kernel.ir_analysis?.io_counts;
+  const ttgir_info = io_counts?.["amd_ttgir_bufferops_count"];
+  const amdgcn_info = io_counts?.["amd_gcn_bufferops_count"];
+  const getCount = (info: Record<string, number> | undefined, key: string): string => { return info?.[key]?.toString() ?? "N/A"; };
+
+  return (
+    <div className="p-6">
+      <h1 className="text-2xl font-bold text-gray-800 mb-6">Triton Kernel IR Analysis</h1>
+
+      <div className="bg-white rounded-lg p-4 mb-4 shadow-sm border border-gray-200">
+        <h2 className="text-xl font-semibold mb-4 text-gray-800">
+          Kernel: {kernel.name}
+        </h2>
+
+        {io_counts && (ttgir_info || amdgcn_info) && (
+          <>
+            <h3 className="text-lg font-medium mb-3 text-gray-800">
+              AMD BufferOps Information
+            </h3>
+
+            <div className="bg-gray-50 p-4 rounded-md border border-gray-200">
+              <div className="grid grid-cols-[repeat(auto-fit,_minmax(180px,_1fr))] gap-3">
+                {ttgir_info && (
+                  <>
+                    <div className="flex flex-col">
+                      <span className="text-sm font-medium text-gray-500">Tiled Buffer Load Count</span>
+                      <span className="font-mono text-sm break-words">{getCount(ttgir_info, "tt.load_count")}</span>
+                    </div>
+                    <div className="flex flex-col">
+                      <span className="text-sm font-medium text-gray-500">Tiled Buffer Store Count</span>
+                      <span className="font-mono text-sm break-words">{getCount(ttgir_info, "tt.store_count")}</span>
+                    </div>
+                    <div className="flex flex-col">
+                      <span className="text-sm font-medium text-gray-500">Tiled Global Load Count</span>
+                      <span className="font-mono text-sm break-words">{getCount(ttgir_info, "amdgpu.buffer_load_count")}</span>
+                    </div>
+                    <div className="flex flex-col">
+                      <span className="text-sm font-medium text-gray-500">Tiled Global Store Count</span>
+                      <span className="font-mono text-sm break-words">{getCount(ttgir_info, "amdgpu.buffer_store_count")}</span>
+                    </div>
+                  </>
+                )}
+                {amdgcn_info && (
+                  <>
+                    <div className="flex flex-col">
+                      <span className="text-sm font-medium text-gray-500">AMDGCN Global Load Instruction Count</span>
+                      <span className="font-mono text-sm break-words">{getCount(amdgcn_info, "global_load_count")}</span>
+                    </div>
+                    <div className="flex flex-col">
+                      <span className="text-sm font-medium text-gray-500">AMDGCN Global Store Instruction Count</span>
+                      <span className="font-mono text-sm break-words">{getCount(amdgcn_info, "global_store_count")}</span>
+                    </div>
+                    <div className="flex flex-col">
+                      <span className="text-sm font-medium text-gray-500">AMDGCN Buffer Load Instruction Count</span>
+                      <span className="font-mono text-sm break-words">{getCount(amdgcn_info, "buffer_load_count")}</span>
+                    </div>
+                    <div className="flex flex-col">
+                      <span className="text-sm font-medium text-gray-500">AMDGCN Buffer Store Instruction Count</span>
+                      <span className="font-mono text-sm break-words">{getCount(amdgcn_info, "buffer_store_count")}</span>
+                    </div>
+                  </>
+                )}
+              </div>
+            </div>
+          </>
+        )}
+      </div>
+    </div>
+  );
+};
+
+export default IRAnalysis;

{tritonparse-0.3.1.dev20251026071454 → tritonparse-0.3.1.dev20251028071524}/website/src/utils/dataLoader.ts

@@ -167,6 +167,11 @@ export interface CompilationMetadata {
     [key: string]: any; // Allow additional unknown fields
 }
 
+export interface IRAnalysisData {
+    // Mapping from IR stage -> <IO type -> count>
+    io_counts?: Record<string, Record<string, number>>;
+}
+
 /**
  * Extracted argument information
  */
@@ -224,6 +229,7 @@ export interface LogEntry {
     launch_index_map?: LaunchRange[];
     diffs?: LaunchDiffData;
     sames?: LaunchSamesData;
+    ir_analysis?: IRAnalysisData; // Stored IR Analysis information.
 }
 
 /**
@@ -239,6 +245,7 @@ export interface ProcessedKernel {
     pythonSourceInfo?: PythonSourceCodeInfo; // Python source code information
     metadata?: KernelMetadata; // Compilation metadata
     launchDiff?: LogEntry; // Aggregated launch event differences
+    ir_analysis?: IRAnalysisData; // Stored IR Analysis information.
 }
 
 /**
@@ -503,6 +510,15 @@ export function processKernelData(logEntries: LogEntry[]): ProcessedKernel[] {
                 console.warn(`Could not find matching kernel for launch_diff hash: ${hash}`);
             }
         }
+        if (entry.event_type === "ir_analysis") {
+            const hash = entry.hash;
+            if (hash && kernelsByHash.has(hash)) {
+                const kernel = kernelsByHash.get(hash)!;
+                kernel.ir_analysis = entry.ir_analysis!; // Attach the ir_analysis
+            } else {
+                console.warn(`Could not find matching kernel for ir_analysis hash: ${hash}`);
+            }
+        }
     }
 
     const finalKernels = Array.from(kernelsByHash.values());

tritonparse-0.3.1.dev20251026071454/website/index.html

@@ -1,14 +0,0 @@
-<!DOCTYPE html>
-<html lang="en">
-  <head>
-    <meta charset="UTF-8" />
-    <link rel="icon" type="image/svg+xml" href="/logo.svg" />
-    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>TritonParse</title>
-  </head>
-  <body>
-    <div id="root"></div>
-    <script type="module" src="/src/main.tsx"></script>
-  </body>
-</html>
-