tritonparse 0.2.1.dev20250913071432__py3-none-any.whl → 0.2.1.dev20250914071450__py3-none-any.whl

tritonparse/common.py

@@ -357,15 +357,13 @@ def parse_logs(
                     file_mapping[rank_key]["mapped_file"] = mapped_file
 
     # Clean up the file mapping - remove None mapped_files and ensure no duplicates
-    for rank_key in file_mapping:
+    for rank_key, rank_data in file_mapping.items():
         if rank_key != "tritonparse_url_prefix":
             # Remove duplicates from regular_files
-            file_mapping[rank_key]["regular_files"] = list(
-                set(file_mapping[rank_key]["regular_files"])
-            )
+            rank_data["regular_files"] = list(set(rank_data["regular_files"]))
             # Remove mapped_file if None
-            if file_mapping[rank_key]["mapped_file"] is None:
-                del file_mapping[rank_key]["mapped_file"]
+            if rank_data["mapped_file"] is None:
+                del rank_data["mapped_file"]
     # Save file mapping to parsed_log_dir
     log_file_list_path = os.path.join(parsed_log_dir, "log_file_list.json")
     with open(log_file_list_path, "w") as f:

{tritonparse-0.2.1.dev20250913071432.dist-info → tritonparse-0.2.1.dev20250914071450.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.2.1.dev20250913071432
+Version: 0.2.1.dev20250914071450
 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
@@ -8,7 +8,10 @@ Project-URL: Homepage, https://github.com/meta-pytorch/tritonparse
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: triton>3.3.1
+Provides-Extra: triton
+Requires-Dist: triton>3.3.1; extra == "triton"
+Provides-Extra: pytorch-triton
+Requires-Dist: pytorch-triton>=3.4.0; extra == "pytorch-triton"
 Provides-Extra: test
 Requires-Dist: coverage>=7.0.0; extra == "test"
 Dynamic: license-file

{tritonparse-0.2.1.dev20250913071432.dist-info → tritonparse-0.2.1.dev20250914071450.dist-info}/RECORD

@@ -1,5 +1,5 @@
 tritonparse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tritonparse/common.py,sha256=VWKFZJG7msIExMH0QNYCO8SHKqUBFtdRuLaCy_K8DFI,13725
+tritonparse/common.py,sha256=aT7zIPKEiuvTq_MbHgMREuVIz-gVcsRskSDlvuIOHuQ,13662
 tritonparse/event_diff.py,sha256=yOD6uNxLJroatfx2nEGr-erw24ObOrHU9P6V5pzr8do,4907
 tritonparse/extract_source_mappings.py,sha256=Z6UxFj2cCE5NCWLQTYPKqUpLfbYhqP8xgCl5mvud9KI,1451
 tritonparse/ir_parser.py,sha256=1j1tP9jpUN7wH3e01bKUkUPgTMlNXUdp8LKRCC-WTro,9324
@@ -17,8 +17,8 @@ tritonparse/tools/format_fix.py,sha256=Ol0Sjui8D7OzHwbamAfGnq8V5Y63uwNaFTKSORN5H
 tritonparse/tools/load_tensor.py,sha256=tfdmNVd9gsZqO6msQBhbXIhOvUzgc83yF64k2GDWPNk,2122
 tritonparse/tools/prettify_ndjson.py,sha256=VOzVWoXpCbaAXYA4i_wBcQIHfh-JhAx7xR4cF_L8yDs,10928
 tritonparse/tools/readme.md,sha256=w6PWYfYnRgoPArLjxG9rVrpcLUkoVMGuRlbpF-o0IQM,110
-tritonparse-0.2.1.dev20250913071432.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
-tritonparse-0.2.1.dev20250913071432.dist-info/METADATA,sha256=XPRwEq-B5zma_jTdr7Gb6G0JZuBA3MN7pXeOYzhn10g,6169
-tritonparse-0.2.1.dev20250913071432.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tritonparse-0.2.1.dev20250913071432.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
-tritonparse-0.2.1.dev20250913071432.dist-info/RECORD,,
+tritonparse-0.2.1.dev20250914071450.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
+tritonparse-0.2.1.dev20250914071450.dist-info/METADATA,sha256=0NjvdtMaAEWf5ro_pBKMHiG_AjdIJ3Zez5js5Yv83zU,6306
+tritonparse-0.2.1.dev20250914071450.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tritonparse-0.2.1.dev20250914071450.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
+tritonparse-0.2.1.dev20250914071450.dist-info/RECORD,,