tritonparse 0.3.1.dev20251028071524__tar.gz → 0.3.1.dev20251029071541__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.3.1.dev20251028071524
+Version: 0.3.1.dev20251029071541
 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

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

@@ -68,12 +68,20 @@ def main():
         }
         unified_parse(**parse_args)
     elif args.func == "reproduce":
+        replacer = None
+        if args.use_fbcode:
+            from tritonparse.fb.reproducer.replacer import FBCodePlaceholderReplacer
+
+            replacer = FBCodePlaceholderReplacer()
+            print(f"Using FBCode placeholder replacer for template: {args.template}")
+
         reproduce(
             input_path=args.input,
             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,
+            replacer=replacer,
         )
     else:
         raise RuntimeError(f"Unknown command: {args.func}")

{tritonparse-0.3.1.dev20251028071524 → tritonparse-0.3.1.dev20251029071541}/tritonparse/reproducer/cli.py

@@ -46,3 +46,8 @@ def _add_reproducer_args(parser: argparse.ArgumentParser) -> None:
             "Defaults to 'default'."
         ),
     )
+    parser.add_argument(
+        "--use-fbcode",
+        action="store_true",
+        help=("Use fbcode to setup repro environment."),
+    )

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

@@ -23,7 +23,7 @@ def reproduce(
     template: str,
     replacer: Optional[PlaceholderReplacer] = None,
     kernel_import: KernelImportMode = KernelImportMode.DEFAULT,
-) -> dict[str, Path]:
+) -> dict[str, str]:
     """
     Generate a reproducer script from NDJSON trace file.
 
@@ -45,7 +45,7 @@ def reproduce(
         f"Built context bundle for kernel: {context_bundle.kernel_info.function_name}"
     )
     out_py_path, temp_json_path = determine_output_paths(
-        out_dir, context_bundle.kernel_info.function_name
+        out_dir, context_bundle.kernel_info.function_name, template
     )
     save_prettified_json(context_bundle.raw_launch_event, temp_json_path)
 

{tritonparse-0.3.1.dev20251028071524 → tritonparse-0.3.1.dev20251029071541}/tritonparse/reproducer/placeholder_replacer.py

@@ -76,21 +76,39 @@ class DefaultPlaceholderReplacer(PlaceholderReplacer):
     - # {{KERNEL_INVOCATION_PLACEHOLDER}}: Replaced with kernel invocation code
     """
 
+    KERNEL_NAME_PLACEHOLDER = "{{KERNEL_NAME_PLACEHOLDER}}"
+    JSON_FILE_NAME_PLACEHOLDER = "{{JSON_FILE_NAME_PLACEHOLDER}}"
+    IR_OVERRIDE_SETUP_PLACEHOLDER = "# {{IR_OVERRIDE_SETUP_PLACEHOLDER}}"
+    KERNEL_SYSPATH_PLACEHOLDER = "# {{KERNEL_SYSPATH_PLACEHOLDER}}"
+    KERNEL_IMPORT_PLACEHOLDER = "# {{KERNEL_IMPORT_PLACEHOLDER}}"
+    UTILITY_FUNCTIONS_PLACEHOLDER = "# {{UTILITY_FUNCTIONS_PLACEHOLDER}}"
+    KERNEL_INVOCATION_PLACEHOLDER = "# {{KERNEL_INVOCATION_PLACEHOLDER}}"
+
     def __init__(self):
         super().__init__()
         # Register all default handlers
-        self.register("{{JSON_FILE_NAME_PLACEHOLDER}}", self._replace_json_filename)
+        self.register(self.JSON_FILE_NAME_PLACEHOLDER, self._replace_json_filename)
         self.register(
-            "# {{IR_OVERRIDE_SETUP_PLACEHOLDER}}", self._replace_ir_override_setup
+            self.IR_OVERRIDE_SETUP_PLACEHOLDER, self._replace_ir_override_setup
         )
-        self.register("# {{KERNEL_SYSPATH_PLACEHOLDER}}", self._replace_kernel_syspath)
-        self.register("# {{KERNEL_IMPORT_PLACEHOLDER}}", self._replace_kernel_import)
+        self.register(self.KERNEL_SYSPATH_PLACEHOLDER, self._replace_kernel_syspath)
+        self.register(self.KERNEL_IMPORT_PLACEHOLDER, self._replace_kernel_import)
         self.register(
-            "# {{UTILITY_FUNCTIONS_PLACEHOLDER}}", self._replace_utility_functions
+            self.UTILITY_FUNCTIONS_PLACEHOLDER, self._replace_utility_functions
         )
         self.register(
-            "# {{KERNEL_INVOCATION_PLACEHOLDER}}", self._replace_kernel_invocation
+            self.KERNEL_INVOCATION_PLACEHOLDER, self._replace_kernel_invocation
         )
+        self.register(self.KERNEL_NAME_PLACEHOLDER, self._replace_kernel_name)
+
+    def _replace_kernel_name(
+        self, code: str, context_bundle: ContextBundle, **kwargs
+    ) -> str:
+        """Replace the kernel name placeholder."""
+        kernel_name = context_bundle.kernel_info.function_name
+        if not kernel_name:
+            raise ValueError("Kernel function name is not available")
+        return code.replace(self.KERNEL_NAME_PLACEHOLDER, kernel_name)
 
     def _replace_json_filename(
         self, code: str, context_bundle: ContextBundle, **kwargs
@@ -99,7 +117,7 @@ class DefaultPlaceholderReplacer(PlaceholderReplacer):
         temp_json_path = kwargs.get("temp_json_path")
         if temp_json_path is None:
             raise ValueError("temp_json_path is required for JSON filename replacement")
-        return code.replace("{{JSON_FILE_NAME_PLACEHOLDER}}", temp_json_path.name)
+        return code.replace(self.JSON_FILE_NAME_PLACEHOLDER, temp_json_path.name)
 
     def _replace_ir_override_setup(
         self, code: str, context_bundle: ContextBundle, **kwargs
@@ -108,7 +126,7 @@ class DefaultPlaceholderReplacer(PlaceholderReplacer):
         kernel_import = kwargs.get("kernel_import", KernelImportMode.DEFAULT)
 
         if kernel_import != KernelImportMode.OVERRIDE_TTIR:
-            return code.replace("# {{IR_OVERRIDE_SETUP_PLACEHOLDER}}", "")
+            return code.replace(self.IR_OVERRIDE_SETUP_PLACEHOLDER, "")
 
         comp_json_filename = kwargs.get("comp_json_filename")
         if not comp_json_filename:
@@ -158,7 +176,7 @@ _original_autotune = triton.autotune
 triton.autotune = _patched_autotune
 '''
 
-        return code.replace("# {{IR_OVERRIDE_SETUP_PLACEHOLDER}}", setup_code)
+        return code.replace(self.IR_OVERRIDE_SETUP_PLACEHOLDER, setup_code)
 
     def _replace_kernel_syspath(
         self, code: str, context_bundle: ContextBundle, **kwargs
@@ -168,15 +186,15 @@ triton.autotune = _patched_autotune
 
         if kernel_import == KernelImportMode.DEFAULT:
             sys_stmt, _ = _generate_import_statements(context_bundle.kernel_info)
-            return code.replace("# {{KERNEL_SYSPATH_PLACEHOLDER}}", sys_stmt)
+            return code.replace(self.KERNEL_SYSPATH_PLACEHOLDER, sys_stmt)
         elif kernel_import == KernelImportMode.COPY:
             comment = (
                 "# Kernel sys.path setup skipped - kernel source code embedded below"
             )
-            return code.replace("# {{KERNEL_SYSPATH_PLACEHOLDER}}", comment)
+            return code.replace(self.KERNEL_SYSPATH_PLACEHOLDER, comment)
         elif kernel_import == KernelImportMode.OVERRIDE_TTIR:
             comment = "# Kernel sys.path setup skipped - using IR override mode"
-            return code.replace("# {{KERNEL_SYSPATH_PLACEHOLDER}}", comment)
+            return code.replace(self.KERNEL_SYSPATH_PLACEHOLDER, comment)
         else:
             raise ValueError(f"Unknown kernel_import mode: {kernel_import}")
 
@@ -190,7 +208,7 @@ triton.autotune = _patched_autotune
             _, import_statement = _generate_import_statements(
                 context_bundle.kernel_info
             )
-            return code.replace("# {{KERNEL_IMPORT_PLACEHOLDER}}", import_statement)
+            return code.replace(self.KERNEL_IMPORT_PLACEHOLDER, import_statement)
         elif kernel_import == KernelImportMode.COPY:
             source_code = context_bundle.kernel_info.source_code
             func_name = context_bundle.kernel_info.function_name
@@ -216,10 +234,10 @@ triton.autotune = _patched_autotune
             embedded_code += "\n" + source_code
             embedded_code += f"\n\n# Use kernel function directly\nimported_kernel_function = {func_name}"
 
-            return code.replace("# {{KERNEL_IMPORT_PLACEHOLDER}}", embedded_code)
+            return code.replace(self.KERNEL_IMPORT_PLACEHOLDER, embedded_code)
         elif kernel_import == KernelImportMode.OVERRIDE_TTIR:
             comment = "# Kernel import skipped - using IR override mode with TTIR"
-            return code.replace("# {{KERNEL_IMPORT_PLACEHOLDER}}", comment)
+            return code.replace(self.KERNEL_IMPORT_PLACEHOLDER, comment)
         else:
             raise ValueError(f"Unknown kernel_import mode: {kernel_import}")
 
@@ -228,7 +246,7 @@ triton.autotune = _patched_autotune
     ) -> str:
         """Replace the utility functions placeholder with extracted functions."""
         utility_code = extract_utility_functions()
-        return code.replace("# {{UTILITY_FUNCTIONS_PLACEHOLDER}}", utility_code)
+        return code.replace(self.UTILITY_FUNCTIONS_PLACEHOLDER, utility_code)
 
     def _replace_kernel_invocation(
         self, code: str, context_bundle: ContextBundle, **kwargs
@@ -237,4 +255,4 @@ triton.autotune = _patched_autotune
         source_code = context_bundle.kernel_info.source_code
         pos_args, kw_args = _parse_kernel_signature(source_code)
         invocation_snippet = _generate_invocation_snippet(pos_args, kw_args)
-        return code.replace("# {{KERNEL_INVOCATION_PLACEHOLDER}}", invocation_snippet)
+        return code.replace(self.KERNEL_INVOCATION_PLACEHOLDER, invocation_snippet)

{tritonparse-0.3.1.dev20251028071524 → tritonparse-0.3.1.dev20251029071541}/tritonparse/reproducer/templates/example.py

@@ -14,7 +14,7 @@ import torch
 # {{UTILITY_FUNCTIONS_PLACEHOLDER}}
 
 
-if __name__ == "__main__":
+def launch_kernel():
     script_dir = Path(__file__).resolve().parent  # noqa: F821
     json_file = script_dir / "{{JSON_FILE_NAME_PLACEHOLDER}}"
     grid, args_dict = create_args_from_json_file(str(json_file))  # noqa: F821
@@ -28,3 +28,7 @@ if __name__ == "__main__":
 
     torch.cuda.synchronize()
     print("Kernel execution finished.")
+
+
+if __name__ == "__main__":
+    launch_kernel()

tritonparse-0.3.1.dev20251029071541/tritonparse/reproducer/templates/tritonbench.py

@@ -0,0 +1,103 @@
+# (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
+
+from pathlib import Path
+from typing import Any, Callable, Dict, Optional, Tuple
+
+import torch
+from tritonbench.utils.triton_op import (
+    BenchmarkOperator,
+    register_benchmark,
+    REGISTERED_X_VALS,
+)
+
+
+imported_kernel_function: Optional[Callable[[Tuple[int], Dict[str, Any]], None]] = None
+
+# {{IR_OVERRIDE_SETUP_PLACEHOLDER}}
+
+# {{KERNEL_SYSPATH_PLACEHOLDER}}
+
+# {{KERNEL_IMPORT_PLACEHOLDER}}
+
+# {{UTILITY_FUNCTIONS_PLACEHOLDER}}
+
+assert imported_kernel_function is not None, "imported_kernel_function is missing"
+
+KERNEL_NAME = "{{KERNEL_NAME_PLACEHOLDER}}"
+REPRO_CONTEXT_FILE_NAME = "{{JSON_FILE_NAME_PLACEHOLDER}}"
+
+
+def _get_launch_kernel_args() -> Tuple[Tuple[int], Dict[str, Any]]:
+    script_dir = Path(__file__).resolve().parent  # noqa: F821
+    json_file = script_dir / REPRO_CONTEXT_FILE_NAME
+
+    grid, args_dict = create_args_from_json_file(json_file)  # noqa: F821, F841
+
+    print("Recorded kernel arguments dictionary:")
+    for name, arg in args_dict.items():
+        if isinstance(arg, torch.Tensor):
+            print(
+                f"  {name}: Tensor:  {arg.shape} {arg.dtype} stride: {arg.stride()}, is_contiguous: {arg.is_contiguous()}"
+            )
+        else:
+            print(f"  {name}: {arg}")
+    print(f"Grid: {grid}")
+
+    return tuple(grid), args_dict
+
+
+grid, args_dict = _get_launch_kernel_args()
+
+
+def _launch_kernel(grid: tuple[int], args_dict: dict[str, Any]):
+    try:
+        assert grid is not None
+        assert args_dict is not None
+
+        # {{KERNEL_INVOCATION_PLACEHOLDER}}
+
+    except Exception as e:
+        print(f"Error: {e}")
+        print("Failed to launch kernel!")
+
+
+# HACK: @register_x_val doesn't allow us to pass `operator_name`` as a parameter
+tensor_args = {k: v for k, v in args_dict.items() if isinstance(v, torch.Tensor)}
+x_vals_label = ", ".join(tensor_args.keys())
+REGISTERED_X_VALS[KERNEL_NAME] = x_vals_label
+
+
+class Operator(BenchmarkOperator):
+    @register_benchmark(operator_name=KERNEL_NAME)
+    def run_kernel(self, grid, args_dict):
+        return lambda: _launch_kernel(grid, args_dict)
+
+    def get_input_iter(self):
+        yield {"grid": grid, "args_dict": args_dict}
+
+    def get_x_val(self, example_inputs):
+        tensors_shapes = [
+            tuple(v.shape)
+            for v in example_inputs["args_dict"].values()
+            if isinstance(v, torch.Tensor)
+        ]
+        return tuple(tensors_shapes)
+
+
+if __name__ == "__main__":
+    print("do_benchmark...")
+
+    args = [
+        "--benchmark-name",
+        KERNEL_NAME,
+    ]
+
+    from tritonbench.utils.parser import get_parser
+
+    parser = get_parser(args)
+    tb_args, extra_args = parser.parse_known_args(args)
+    bench = Operator(tb_args, extra_args)
+    bench.run()
+
+    print(bench.output)
+    print("Benchmark completed successfully!")

{tritonparse-0.3.1.dev20251028071524 → tritonparse-0.3.1.dev20251029071541}/tritonparse/reproducer/utils.py

@@ -327,7 +327,7 @@ def _create_arg_from_info(arg_info):
         return None
 
 
-def determine_output_paths(out_dir: str, kernel_name: str):
+def determine_output_paths(out_dir: str, kernel_name: str, template: str):
     """
     Determine output file paths for reproducer script and context data.
 
@@ -342,7 +342,12 @@ def determine_output_paths(out_dir: str, kernel_name: str):
     output_directory = Path(out_dir) / kernel_name
     output_directory.mkdir(parents=True, exist_ok=True)
 
-    out_py_path = output_directory / f"repro_{timestamp}.py"
+    filename_parts = ["repro"]
+    if template != "example":
+        filename_parts.append(template.replace(".", "_"))
+    filename_parts.append(timestamp)
+    filename = "_".join(filename_parts) + ".py"
+    out_py_path = output_directory / filename
     temp_json_path = output_directory / f"repro_context_{timestamp}.json"
 
     return out_py_path, temp_json_path

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.3.1.dev20251028071524
+Version: 0.3.1.dev20251029071541
 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

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

@@ -62,6 +62,7 @@ tritonparse/reproducer/ingestion/ndjson.py
 tritonparse/reproducer/templates/__init__.py
 tritonparse/reproducer/templates/example.py
 tritonparse/reproducer/templates/loader.py
+tritonparse/reproducer/templates/tritonbench.py
 tritonparse/tools/__init__.py
 tritonparse/tools/decompress_bin_ndjson.py
 tritonparse/tools/disasm.py
@@ -92,6 +93,7 @@ website/src/assets/react.svg
 website/src/components/ArgumentViewer.tsx
 website/src/components/Callstack.tsx
 website/src/components/CodeComparisonView.tsx
+website/src/components/CodeViewer.css
 website/src/components/CodeViewer.tsx
 website/src/components/CompilationInfo.tsx
 website/src/components/CopyCodeButton.tsx

tritonparse-0.3.1.dev20251029071541/website/src/components/CodeViewer.css

@@ -0,0 +1,43 @@
+/* Location definition line styles */
+.loc-definition-line {
+    background-color: rgba(255, 215, 0, 0.1) !important;
+    border-left: 3px solid #ffd700 !important;
+}
+
+.loc-definition-line:hover {
+    background-color: rgba(255, 215, 0, 0.2) !important;
+}
+
+/* Alias name badge */
+.alias-badge {
+    display: inline-block;
+    margin-left: 8px;
+    padding: 2px 6px;
+    background: #e3f2fd;
+    border-radius: 3px;
+    font-size: 0.85em;
+    color: #1976d2;
+    font-weight: 500;
+    vertical-align: middle;
+}
+
+.alias-badge:hover {
+    background: #bbdefb;
+}
+
+/* Dark theme adjustments */
+[data-theme="dark"] .alias-badge {
+    background: #1565c0;
+    color: #e3f2fd;
+}
+
+[data-theme="dark"] .alias-badge:hover {
+    background: #1976d2;
+}
+
+/* Loc ID indicator */
+.loc-id-indicator {
+    opacity: 0.6;
+    font-size: 0.8em;
+    margin-left: 4px;
+}

{tritonparse-0.3.1.dev20251028071524 → tritonparse-0.3.1.dev20251029071541}/website/src/components/CodeViewer.tsx

@@ -5,6 +5,7 @@ import {
   oneDark,
 } from "react-syntax-highlighter/dist/esm/styles/prism";
 import type { SourceMapping } from "../utils/dataLoader";
+import "./CodeViewer.css";
 
 // Import language support
 import llvm from 'react-syntax-highlighter/dist/esm/languages/prism/llvm';
@@ -207,6 +208,7 @@ const BasicCodeViewer: React.FC<CodeViewerProps> = ({
   highlightedLines = [],
   onLineClick,
   viewerId,
+  sourceMapping,
 }) => {
   const containerRef = useRef<HTMLDivElement>(null);
   const lines = splitIntoLines(code);
@@ -239,6 +241,21 @@ const BasicCodeViewer: React.FC<CodeViewerProps> = ({
           {lines.map((line, index) => {
             const lineNumber = index + 1;
             const isHighlighted = highlightedLines.includes(lineNumber);
+            const mapping = sourceMapping?.[lineNumber.toString()];
+            const isLocDef = mapping?.type === "loc_def" || mapping?.kind === "loc_def";
+            
+            // Build tooltip text for loc definitions
+            let tooltipText = "";
+            if (isLocDef && mapping) {
+              tooltipText = `Location definition`;
+              if (mapping.alias_name) {
+                tooltipText += `: ${mapping.alias_name}`;
+              }
+              if (mapping.alias_of !== undefined) {
+                const targetKey = mapping.alias_of === "" ? "#loc" : `#loc${mapping.alias_of}`;
+                tooltipText += ` → ${targetKey}`;
+              }
+            }
 
             return (
               <div
@@ -255,7 +272,11 @@ const BasicCodeViewer: React.FC<CodeViewerProps> = ({
                 }}
                 onClick={() => handleLineClick(lineNumber)}
                 data-line-number={lineNumber}
-                className={isHighlighted ? 'highlighted-line' : ''}
+                className={`${isHighlighted ? 'highlighted-line' : ''} ${isLocDef ? 'loc-definition-line' : ''}`}
+                title={tooltipText}
+                data-loc-id={mapping?.loc_id}
+                data-alias-name={mapping?.alias_name}
+                data-loc-def={isLocDef ? "true" : undefined}
               >
                 <span style={{
                   position: "absolute",
@@ -290,6 +311,7 @@ const LargeFileViewer: React.FC<CodeViewerProps> = ({
   highlightedLines = [],
   onLineClick,
   viewerId,
+  sourceMapping,
 }) => {
   const containerRef = useRef<HTMLDivElement>(null);
   const [visibleRange, setVisibleRange] = useState({ start: 0, end: 50 });
@@ -421,6 +443,8 @@ const LargeFileViewer: React.FC<CodeViewerProps> = ({
             lineProps={(lineNumber) => {
               // Adjust line number based on visible range
               const actualLine = lineNumber + visibleRange.start;
+              const mapping = sourceMapping?.[actualLine.toString()];
+              const isLocDef = mapping?.type === "loc_def" || mapping?.kind === "loc_def";
 
               // Create styles for the line
               const style: React.CSSProperties = {
@@ -440,11 +464,28 @@ const LargeFileViewer: React.FC<CodeViewerProps> = ({
 
               }
 
+              // Build tooltip text for loc definitions
+              let tooltipText = "";
+              if (isLocDef && mapping) {
+                tooltipText = `Location definition`;
+                if (mapping.alias_name) {
+                  tooltipText += `: ${mapping.alias_name}`;
+                }
+                if (mapping.alias_of !== undefined) {
+                  const targetKey = mapping.alias_of === "" ? "#loc" : `#loc${mapping.alias_of}`;
+                  tooltipText += ` → ${targetKey}`;
+                }
+              }
+
               return {
                 style,
                 onClick: () => handleLineClick(actualLine),
                 'data-line-number': actualLine,
-                className: isHighlighted ? 'highlighted-line' : '',
+                className: `${isHighlighted ? 'highlighted-line' : ''} ${isLocDef ? 'loc-definition-line' : ''}`,
+                title: tooltipText,
+                'data-loc-id': mapping?.loc_id,
+                'data-alias-name': mapping?.alias_name,
+                'data-loc-def': isLocDef ? "true" : undefined,
               };
             }}
             customStyle={{
@@ -474,6 +515,7 @@ const StandardCodeViewer: React.FC<CodeViewerProps> = ({
   highlightedLines = [],
   onLineClick,
   viewerId,
+  sourceMapping,
 }) => {
   const containerRef = useRef<HTMLDivElement>(null);
 
@@ -537,6 +579,22 @@ const StandardCodeViewer: React.FC<CodeViewerProps> = ({
         wrapLines={true}
         lineProps={(lineNumber) => {
           const isHighlighted = highlightedLines.includes(lineNumber);
+          const mapping = sourceMapping?.[lineNumber.toString()];
+          const isLocDef = mapping?.type === "loc_def" || mapping?.kind === "loc_def";
+          
+          // Build tooltip text for loc definitions
+          let tooltipText = "";
+          if (isLocDef && mapping) {
+            tooltipText = `Location definition`;
+            if (mapping.alias_name) {
+              tooltipText += `: ${mapping.alias_name}`;
+            }
+            if (mapping.alias_of !== undefined) {
+              const targetKey = mapping.alias_of === "" ? "#loc" : `#loc${mapping.alias_of}`;
+              tooltipText += ` → ${targetKey}`;
+            }
+          }
+          
           return {
             style: {
               backgroundColor: isHighlighted
@@ -549,7 +607,11 @@ const StandardCodeViewer: React.FC<CodeViewerProps> = ({
             },
             onClick: () => handleLineClick(lineNumber),
             "data-line-number": lineNumber,
-            className: isHighlighted ? 'highlighted-line' : '',
+            className: `${isHighlighted ? 'highlighted-line' : ''} ${isLocDef ? 'loc-definition-line' : ''}`,
+            title: tooltipText,
+            "data-loc-id": mapping?.loc_id,
+            "data-alias-name": mapping?.alias_name,
+            "data-loc-def": isLocDef ? "true" : undefined,
           };
         }}
       >

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

@@ -17,6 +17,12 @@ export interface SourceMapping {
     ttgir_lines?: number[]; // Array of corresponding TTGIR lines
     llir_lines?: number[]; // Array of corresponding LLIR lines
     amdgcn_lines?: number[]; // Array of corresponding AMDGCN lines
+    // New fields for location alias support
+    type?: string; // Type of mapping entry, e.g., "loc_def" for loc definition lines
+    kind?: string; // Deprecated alias for type, kept for backward compatibility
+    loc_id?: string; // The #loc identifier (e.g., "13" for #loc13)
+    alias_name?: string; // Name of the alias (e.g., "x_ptr" in #loc13 = loc("x_ptr"(#loc)))
+    alias_of?: string; // The target #loc this alias points to
 }
 
 /**

{tritonparse-0.3.1.dev20251028071524 → tritonparse-0.3.1.dev20251029071541}/website/src/pages/IRAnalysis.tsx

@@ -49,19 +49,19 @@ const IRAnalysis: React.FC<IRAnalysisProps> = ({ kernels, selectedKernel }) => {
                 {ttgir_info && (
                   <>
                     <div className="flex flex-col">
-                      <span className="text-sm font-medium text-gray-500">Tiled Buffer Load Count</span>
+                      <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, "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="text-sm font-medium text-gray-500">Tiled Global 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="text-sm font-medium text-gray-500">Tiled Buffer 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="text-sm font-medium text-gray-500">Tiled Buffer Store Count</span>
                       <span className="font-mono text-sm break-words">{getCount(ttgir_info, "amdgpu.buffer_store_count")}</span>
                     </div>
                   </>