tritonparse 0.2.1.dev20250916071516__py3-none-any.whl → 0.2.1.dev20250918071513__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of tritonparse might be problematic. Click here for more details.

@@ -555,7 +555,12 @@ class TritonTraceHandler(logging.StreamHandler):
555
555
  )
556
556
  should_set_root_dir = False
557
557
  # TODO: change to tritonparse knob
558
- elif not torch._utils_internal.justknobs_check("pytorch/trace:enable"):
558
+ # The following check is necessary because the possible version mismatch between torch and tritonparse
559
+ elif (
560
+ hasattr(torch, "_utils_internal")
561
+ and hasattr(torch._utils_internal, "justknobs_check")
562
+ and not torch._utils_internal.justknobs_check("pytorch/trace:enable")
563
+ ):
559
564
  log.info(
560
565
  "TritonTraceHandler: disabled because justknobs_check('pytorch/trace:enable') returned False"
561
566
  )
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: tritonparse
3
- Version: 0.2.1.dev20250916071516
3
+ Version: 0.2.1.dev20250918071513
4
4
  Summary: TritonParse: A Compiler Tracer, Visualizer, and mini-Reproducer Generator for Triton Kernels
5
5
  Author-email: Yueming Hao <yhao@meta.com>
6
6
  License-Expression: BSD-3-Clause
@@ -21,7 +21,7 @@ Dynamic: license-file
21
21
  [![License: BSD-3](https://img.shields.io/badge/License-BSD--3-blue.svg)](https://opensource.org/licenses/BSD-3-Clause)
22
22
  [![GitHub Pages](https://img.shields.io/badge/GitHub%20Pages-Deploy-brightgreen)](https://meta-pytorch.org/tritonparse/)
23
23
 
24
- **A comprehensive visualization and analysis tool for Triton IR files** — helping developers analyze, debug, and understand Triton kernel compilation processes.
24
+ **A comprehensive visualization and analysis tool for Triton kernel compilation and launch** — helping developers analyze, debug, and understand Triton kernel compilation processes.
25
25
 
26
26
  🌐 **[Try it online →](https://meta-pytorch.org/tritonparse/?json_url=https://meta-pytorch.org/tritonparse/dedicated_log_triton_trace_findhao__mapped.ndjson.gz)**
27
27
 
@@ -81,10 +81,18 @@ INFO:tritonparse:Copying parsed logs from /tmp/tmp1gan7zky to /scratch/findhao/t
81
81
  ## 🛠️ Installation
82
82
 
83
83
  **For basic usage (trace generation):**
84
+ Four options:
84
85
  ```bash
86
+ # install nightly version
87
+ pip install -U --pre tritonparse
88
+ # install stable version
89
+ pip install tritonparse
90
+ # install from source
85
91
  git clone https://github.com/meta-pytorch/tritonparse.git
86
92
  cd tritonparse
87
93
  pip install -e .
94
+ # pip install the latest version from github
95
+ pip install git+https://github.com/meta-pytorch/tritonparse.git
88
96
  ```
89
97
 
90
98
  **Prerequisites:** Python ≥ 3.10, Triton ≥ 3.4.0, GPU required (NVIDIA/AMD)
@@ -7,7 +7,7 @@ tritonparse/mapper.py,sha256=prrczfi13P7Aa042OrEBsmRF1HW3jDhwxicANgPkWIM,4150
7
7
  tritonparse/shared_vars.py,sha256=-c9CvXJSDm9spYhDOJPEQProeT_xl3PaNmqTEYi_u4s,505
8
8
  tritonparse/source_type.py,sha256=nmYEQS8rfkIN9BhNhQbkmEvKnvS-3zAxRGLY4TaZdi8,1676
9
9
  tritonparse/sourcemap_utils.py,sha256=qsQmTDuEe9yuUVyxSHRbjTR38gi0hvJEijnPkrJVAV4,2037
10
- tritonparse/structured_logging.py,sha256=7wSKc9HnV40eLoOgFNosc7dXWvSxsJJo5mnRVoDZqIk,43678
10
+ tritonparse/structured_logging.py,sha256=7r9pv6miUdb8-CCZfj8SkD3XItzwPeONmszEL7TZak4,43949
11
11
  tritonparse/tp_logger.py,sha256=vXzY7hMDmVnRBGBhIjFZe3nHZzG5NKKPONGUszJhGgU,242
12
12
  tritonparse/trace_processor.py,sha256=QzUOKwnOkBbwTTKBsa5ZMUABPLMJIBFtTcG2SkhO0I8,12771
13
13
  tritonparse/utils.py,sha256=wt61tpbkqjGqHh0c7Nr2WlOv7PbQssmjULd6uA6aAko,4475
@@ -19,8 +19,8 @@ tritonparse/tools/format_fix.py,sha256=Ol0Sjui8D7OzHwbamAfGnq8V5Y63uwNaFTKSORN5H
19
19
  tritonparse/tools/load_tensor.py,sha256=tfdmNVd9gsZqO6msQBhbXIhOvUzgc83yF64k2GDWPNk,2122
20
20
  tritonparse/tools/prettify_ndjson.py,sha256=VOzVWoXpCbaAXYA4i_wBcQIHfh-JhAx7xR4cF_L8yDs,10928
21
21
  tritonparse/tools/readme.md,sha256=w6PWYfYnRgoPArLjxG9rVrpcLUkoVMGuRlbpF-o0IQM,110
22
- tritonparse-0.2.1.dev20250916071516.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
23
- tritonparse-0.2.1.dev20250916071516.dist-info/METADATA,sha256=5ic4t2gOHTBKvj4oZlYVOERcUOIaO93Ufkbyu6NG5aI,6306
24
- tritonparse-0.2.1.dev20250916071516.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
25
- tritonparse-0.2.1.dev20250916071516.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
26
- tritonparse-0.2.1.dev20250916071516.dist-info/RECORD,,
22
+ tritonparse-0.2.1.dev20250918071513.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
23
+ tritonparse-0.2.1.dev20250918071513.dist-info/METADATA,sha256=d13RR7shdn1sAmyC_1abuKe3yBe0SSE9w4GxnXomwcc,6580
24
+ tritonparse-0.2.1.dev20250918071513.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
25
+ tritonparse-0.2.1.dev20250918071513.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
26
+ tritonparse-0.2.1.dev20250918071513.dist-info/RECORD,,