tritonparse 0.2.1.dev20250917071511__py3-none-any.whl → 0.2.1.dev20250919071533__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.
@@ -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.dev20250917071511
3
+ Version: 0.2.1.dev20250919071533
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
@@ -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.dev20250917071511.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
23
- tritonparse-0.2.1.dev20250917071511.dist-info/METADATA,sha256=nesWLDcKVWKWtq3Wph_hHxTH5abc2SREFy_tYDeACdk,6580
24
- tritonparse-0.2.1.dev20250917071511.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
25
- tritonparse-0.2.1.dev20250917071511.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
26
- tritonparse-0.2.1.dev20250917071511.dist-info/RECORD,,
22
+ tritonparse-0.2.1.dev20250919071533.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
23
+ tritonparse-0.2.1.dev20250919071533.dist-info/METADATA,sha256=Di-voYjuj_B-FcT8AzDpc3zYi0N5dPeNP-rLw5xi8fw,6580
24
+ tritonparse-0.2.1.dev20250919071533.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
25
+ tritonparse-0.2.1.dev20250919071533.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
26
+ tritonparse-0.2.1.dev20250919071533.dist-info/RECORD,,