tritonparse 0.3.1.dev20251024071516__py3-none-any.whl → 0.3.1.dev20251025071457__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.

Files changed (28) hide show
  1. tritonparse/__main__.py +2 -0
  2. tritonparse/context_manager.py +2 -0
  3. tritonparse/event_diff.py +2 -0
  4. tritonparse/ir_analysis.py +2 -0
  5. tritonparse/ir_parser.py +2 -0
  6. tritonparse/mapper.py +2 -0
  7. tritonparse/reproducer/cli.py +2 -0
  8. tritonparse/reproducer/function_extractor.py +2 -0
  9. tritonparse/reproducer/ingestion/ndjson.py +2 -0
  10. tritonparse/reproducer/orchestrator.py +2 -0
  11. tritonparse/reproducer/placeholder_replacer.py +2 -0
  12. tritonparse/reproducer/templates/loader.py +2 -0
  13. tritonparse/reproducer/types.py +2 -0
  14. tritonparse/reproducer/utils.py +2 -0
  15. tritonparse/shared_vars.py +1 -0
  16. tritonparse/sourcemap_utils.py +2 -0
  17. tritonparse/tools/decompress_bin_ndjson.py +2 -0
  18. tritonparse/tools/format_fix.py +2 -0
  19. tritonparse/tools/load_tensor.py +2 -0
  20. tritonparse/tools/prettify_ndjson.py +2 -0
  21. tritonparse/trace_processor.py +2 -0
  22. {tritonparse-0.3.1.dev20251024071516.dist-info → tritonparse-0.3.1.dev20251025071457.dist-info}/METADATA +1 -1
  23. tritonparse-0.3.1.dev20251025071457.dist-info/RECORD +41 -0
  24. tritonparse-0.3.1.dev20251024071516.dist-info/RECORD +0 -41
  25. {tritonparse-0.3.1.dev20251024071516.dist-info → tritonparse-0.3.1.dev20251025071457.dist-info}/WHEEL +0 -0
  26. {tritonparse-0.3.1.dev20251024071516.dist-info → tritonparse-0.3.1.dev20251025071457.dist-info}/entry_points.txt +0 -0
  27. {tritonparse-0.3.1.dev20251024071516.dist-info → tritonparse-0.3.1.dev20251025071457.dist-info}/licenses/LICENSE +0 -0
  28. {tritonparse-0.3.1.dev20251024071516.dist-info → tritonparse-0.3.1.dev20251025071457.dist-info}/top_level.txt +0 -0
tritonparse/__main__.py CHANGED
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  from .cli import main
2
4
 
3
5
 
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import os
2
4
  import shutil
3
5
  import tempfile
tritonparse/event_diff.py CHANGED
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import json
2
4
  from collections import defaultdict
3
5
  from typing import Any, Dict, List, Tuple
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import logging
2
4
 
3
5
  from .sourcemap_utils import load_ir_contents
tritonparse/ir_parser.py CHANGED
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import logging
2
4
  import os
3
5
  import re
tritonparse/mapper.py CHANGED
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import logging
2
4
  from collections import defaultdict
3
5
  from typing import Any, Dict, List, Tuple
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import argparse
2
4
 
3
5
  from tritonparse.reproducer.types import KernelImportMode
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  """
2
4
  Function extractor for reproducer utility functions.
3
5
 
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  from dataclasses import dataclass
2
4
  from typing import Any, Dict, List, Optional, Tuple
3
5
 
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  from pathlib import Path
2
4
  from typing import Optional
3
5
 
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  from abc import ABC
2
4
 
3
5
  from typing import Any, Dict, Protocol
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  from importlib.resources import files as pkg_files
2
4
  from pathlib import Path
3
5
  from typing import List
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  from enum import Enum
2
4
 
3
5
 
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import importlib
2
4
  import importlib.util
3
5
  import json
@@ -1,3 +1,4 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
1
2
  # We'd like to sperate structured logging module and tritonparse module as much as possible. So, put the shared variables here.
2
3
  import os
3
4
 
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import logging
2
4
  from typing import Any, Dict, List
3
5
 
@@ -1,4 +1,6 @@
1
1
  #!/usr/bin/env python3
2
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
3
+
2
4
  """
3
5
  Script to decompress .bin.ndjson files back to regular .ndjson format.
4
6
 
@@ -1,4 +1,6 @@
1
1
  #!/usr/bin/env python3
2
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
3
+
2
4
  """
3
5
  Format fix script for tritonparse project.
4
6
 
@@ -1,4 +1,6 @@
1
1
  #!/usr/bin/env python3
2
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
3
+
2
4
  """
3
5
  Simple tensor loading utility for tritonparse saved tensors.
4
6
  Usage:
@@ -1,4 +1,6 @@
1
1
  #!/usr/bin/env python3
2
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
3
+
2
4
  """
3
5
  Convert an NDJSON file to a prettified JSON file.
4
6
 
@@ -1,3 +1,5 @@
1
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
2
+
1
3
  import gzip
2
4
  import json
3
5
  import logging
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: tritonparse
3
- Version: 0.3.1.dev20251024071516
3
+ Version: 0.3.1.dev20251025071457
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
@@ -0,0 +1,41 @@
1
+ tritonparse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
+ tritonparse/__main__.py,sha256=RXbkALBewcb1xlJBnsQl9IaBRUNln7U8NuRZKT8UdIk,117
3
+ tritonparse/cli.py,sha256=w_sCIv1NlcGDzoflsYuTvtQis92_YuZEC51ZlTVsYCU,2517
4
+ tritonparse/common.py,sha256=MJo9bVCgSKkwXpEoUkUczPo_5jOYpJgXLq4UsWYqN3c,13924
5
+ tritonparse/context_manager.py,sha256=OdMn11qbApYL2c9IlbUpcT27r04ZSa4DfvrY2mLA958,2243
6
+ tritonparse/event_diff.py,sha256=USCjfjYr-7Ie-EfZgtCFMZMA1KRzFRDe7yDFy98zYI4,4962
7
+ tritonparse/extract_source_mappings.py,sha256=Z6UxFj2cCE5NCWLQTYPKqUpLfbYhqP8xgCl5mvud9KI,1451
8
+ tritonparse/ir_analysis.py,sha256=YObFsRKkpdjkwL3lIeWRXU6DY13fGVqdh8li4PiCpoQ,2482
9
+ tritonparse/ir_parser.py,sha256=Qtb8kA6IMBtbLbABEGSDFdGQQI_GsPd1iC2GHbEviaI,9379
10
+ tritonparse/mapper.py,sha256=QBCUMHM9pu3x3ahFp0wyXJbmv9TFGVPdkcLULok1E-k,4205
11
+ tritonparse/shared_vars.py,sha256=RifXq55KisHgspYAmGcaCWY6ZHX8iejFHvwIewvcWZE,707
12
+ tritonparse/source_type.py,sha256=nmYEQS8rfkIN9BhNhQbkmEvKnvS-3zAxRGLY4TaZdi8,1676
13
+ tritonparse/sourcemap_utils.py,sha256=uI02n5Sgnlx7Nc15QAX5N6_tZZMips0PyJuo1n3eouY,2654
14
+ tritonparse/structured_logging.py,sha256=L1xkkCx8Jr9YQbM0Kgtf2g6L3aWMkYOEeFFEOSo8Lkk,60306
15
+ tritonparse/tp_logger.py,sha256=vXzY7hMDmVnRBGBhIjFZe3nHZzG5NKKPONGUszJhGgU,242
16
+ tritonparse/trace_processor.py,sha256=cJHagBOSHFpvUKYLxrvtgoFODDZ5CEFkU_xZTKsD9s8,12981
17
+ tritonparse/utils.py,sha256=Jnlptcd79llSDev-_1XyyOnv2izUqv0PEL74A8GF2tc,4565
18
+ tritonparse/reproducer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
19
+ tritonparse/reproducer/cli.py,sha256=wk0K8qJhvP9gty2EBMH3WEc3TSFcszvNq3JbfYu_sTw,1577
20
+ tritonparse/reproducer/function_extractor.py,sha256=kQr10JKHy8EvAN7ic4Azjz6TYe-udBW2DVmbQ--c1pc,6643
21
+ tritonparse/reproducer/orchestrator.py,sha256=h55xXEVYtIr33GvwCuvM03bsYc01_yPa5X_B-3w71kE,3237
22
+ tritonparse/reproducer/placeholder_replacer.py,sha256=ARPZAa9A3Fyit_dIclOKe1JzFgUPBFdHvfy3z20x2E8,9607
23
+ tritonparse/reproducer/types.py,sha256=86wql3NaGgpkOzx0gDFb5qexNjKExzhL0uIwGU7grrw,564
24
+ tritonparse/reproducer/utils.py,sha256=yFS1Mg2IhRgW-1UNfqjWH5gRSqc8Wbn5Ykre8L-EWcU,16599
25
+ tritonparse/reproducer/ingestion/ndjson.py,sha256=QX_4XyK_vhSOFyJtfDxQNwzXqVXolrfk4cefzlShFps,7427
26
+ tritonparse/reproducer/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
27
+ tritonparse/reproducer/templates/example.py,sha256=jR3c8_d7fAFJYaj1DuUuthnI4Xd-_606bWDRdUPMNyo,785
28
+ tritonparse/reproducer/templates/loader.py,sha256=x14KHXkovOIcXFKii3Jx4XjpEhXqUMqp575qAffi370,1975
29
+ tritonparse/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
30
+ tritonparse/tools/decompress_bin_ndjson.py,sha256=Gn5foDIlxBN5D5wmcdrEmwvxo3_wRlH8ih2U2Ys3RdM,4199
31
+ tritonparse/tools/disasm.py,sha256=c4HmNNoPPeXPQBQkPVcMaHwDHbHNZNxuqXn4UIIs1Z0,2434
32
+ tritonparse/tools/format_fix.py,sha256=ISalg_N_L7Xktag3mLr-G9T6Opxv793s1WG6A9wUtXk,3922
33
+ tritonparse/tools/load_tensor.py,sha256=7-LbpboKDNJFBLNhiKS3enoqRlVAb55OjPc70PwHXAw,2789
34
+ tritonparse/tools/prettify_ndjson.py,sha256=kR8hmBCv-iJeuzpi2_6CZv9T4_edRQbBOSOPpMm6wrw,11117
35
+ tritonparse/tools/readme.md,sha256=w6PWYfYnRgoPArLjxG9rVrpcLUkoVMGuRlbpF-o0IQM,110
36
+ tritonparse-0.3.1.dev20251025071457.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
37
+ tritonparse-0.3.1.dev20251025071457.dist-info/METADATA,sha256=AOVS0JPzUdhi5-59slIrr7oVO0Ofy_8xNQW75pbLj04,8278
38
+ tritonparse-0.3.1.dev20251025071457.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
39
+ tritonparse-0.3.1.dev20251025071457.dist-info/entry_points.txt,sha256=wEXdaieDoRRCCdhEv2p_C68iytnaXU_2pwt5CqjfbWY,56
40
+ tritonparse-0.3.1.dev20251025071457.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
41
+ tritonparse-0.3.1.dev20251025071457.dist-info/RECORD,,
@@ -1,41 +0,0 @@
1
- tritonparse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
- tritonparse/__main__.py,sha256=wu5N2wk8mvBgyvr2ghmQf4prezAe0_i-p123VVreyYc,62
3
- tritonparse/cli.py,sha256=w_sCIv1NlcGDzoflsYuTvtQis92_YuZEC51ZlTVsYCU,2517
4
- tritonparse/common.py,sha256=MJo9bVCgSKkwXpEoUkUczPo_5jOYpJgXLq4UsWYqN3c,13924
5
- tritonparse/context_manager.py,sha256=R6zKoixi8aXF2yK8pN1EUVPPmldhhXMdEK_PgKZjaX4,2188
6
- tritonparse/event_diff.py,sha256=yOD6uNxLJroatfx2nEGr-erw24ObOrHU9P6V5pzr8do,4907
7
- tritonparse/extract_source_mappings.py,sha256=Z6UxFj2cCE5NCWLQTYPKqUpLfbYhqP8xgCl5mvud9KI,1451
8
- tritonparse/ir_analysis.py,sha256=ERu6EWotqcW99ZGNEH-mxGyYej45yt0IXx0OCab_bfw,2427
9
- tritonparse/ir_parser.py,sha256=1j1tP9jpUN7wH3e01bKUkUPgTMlNXUdp8LKRCC-WTro,9324
10
- tritonparse/mapper.py,sha256=prrczfi13P7Aa042OrEBsmRF1HW3jDhwxicANgPkWIM,4150
11
- tritonparse/shared_vars.py,sha256=fCAW24mx9nENYoNbTy-tZjiN5-w6oGTO_av-Pw1J1TY,653
12
- tritonparse/source_type.py,sha256=nmYEQS8rfkIN9BhNhQbkmEvKnvS-3zAxRGLY4TaZdi8,1676
13
- tritonparse/sourcemap_utils.py,sha256=f36lKjc23qx4aOgpHOOo2lxPF4rwgSrOcULZqyUKxn0,2599
14
- tritonparse/structured_logging.py,sha256=L1xkkCx8Jr9YQbM0Kgtf2g6L3aWMkYOEeFFEOSo8Lkk,60306
15
- tritonparse/tp_logger.py,sha256=vXzY7hMDmVnRBGBhIjFZe3nHZzG5NKKPONGUszJhGgU,242
16
- tritonparse/trace_processor.py,sha256=DMY03eqQbM1oUbQuMkfHV_rlaKmW4ggJ048FMCbOH3c,12926
17
- tritonparse/utils.py,sha256=Jnlptcd79llSDev-_1XyyOnv2izUqv0PEL74A8GF2tc,4565
18
- tritonparse/reproducer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
19
- tritonparse/reproducer/cli.py,sha256=MqYuuAP-uAIWWLQxixwyyBHJGSaQdG3xXGaVjERTLX8,1522
20
- tritonparse/reproducer/function_extractor.py,sha256=pB5b52Xlk9-fe-Gs-z_Rn2HkZd1bC9qHTw1JZc_epBM,6588
21
- tritonparse/reproducer/orchestrator.py,sha256=ZGdsiOZg2Bg6o0cqZSCGx_v6mdzR4yZX7S3SVQjZ9-c,3182
22
- tritonparse/reproducer/placeholder_replacer.py,sha256=TvqQIrOubmBHJ_pl0ZvkpP4dIDiMYxUll1q9MIbzZco,9552
23
- tritonparse/reproducer/types.py,sha256=AfVl83zoJZQ58JJoplCcMC51gK-M-OKcafatYEIGgW0,509
24
- tritonparse/reproducer/utils.py,sha256=LHmkM9EEAn9BwNyhHuCEp1tm7omeILbhkAcOmN-iLrk,16544
25
- tritonparse/reproducer/ingestion/ndjson.py,sha256=pEujTl5xXW2E2DEW8ngxXQ8qP9oawb90wBVTWHDs1jk,7372
26
- tritonparse/reproducer/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
27
- tritonparse/reproducer/templates/example.py,sha256=jR3c8_d7fAFJYaj1DuUuthnI4Xd-_606bWDRdUPMNyo,785
28
- tritonparse/reproducer/templates/loader.py,sha256=HqjfThdDVg7q2bYWry78sIaVRkUpkcA8KQDt83YrlVE,1920
29
- tritonparse/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
30
- tritonparse/tools/decompress_bin_ndjson.py,sha256=kpt7DM_sSA334F1X45xdkP2OR9LuB27Pc50EkGr6CPM,4144
31
- tritonparse/tools/disasm.py,sha256=c4HmNNoPPeXPQBQkPVcMaHwDHbHNZNxuqXn4UIIs1Z0,2434
32
- tritonparse/tools/format_fix.py,sha256=Ol0Sjui8D7OzHwbamAfGnq8V5Y63uwNaFTKSORN5HkQ,3867
33
- tritonparse/tools/load_tensor.py,sha256=94-TiSYlpXJx4MPmGK1ovmZlTt56Q_B3KQeCPaA6Cnw,2734
34
- tritonparse/tools/prettify_ndjson.py,sha256=r2YlHwFDTHgML7KljRmMsHaDg29q8gOQAgyDKWJhxRM,11062
35
- tritonparse/tools/readme.md,sha256=w6PWYfYnRgoPArLjxG9rVrpcLUkoVMGuRlbpF-o0IQM,110
36
- tritonparse-0.3.1.dev20251024071516.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
37
- tritonparse-0.3.1.dev20251024071516.dist-info/METADATA,sha256=zSXJHPi5BEV2zS_aiZBVSEGQXY8_xLp1f6YoC2NHFgQ,8278
38
- tritonparse-0.3.1.dev20251024071516.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
39
- tritonparse-0.3.1.dev20251024071516.dist-info/entry_points.txt,sha256=wEXdaieDoRRCCdhEv2p_C68iytnaXU_2pwt5CqjfbWY,56
40
- tritonparse-0.3.1.dev20251024071516.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
41
- tritonparse-0.3.1.dev20251024071516.dist-info/RECORD,,