tritonparse 0.2.4.dev20250930071536__py3-none-any.whl → 0.2.4.dev20251001071529__py3-none-any.whl

tritonparse/reproducer/orchestrator.py

@@ -18,7 +18,7 @@ def reproduce(
     line_index: int,
     out_dir: str,
     template: str,
-):
+) -> dict[str, Path]:
     """
     Generate a reproducer script from NDJSON trace file.
 
@@ -55,9 +55,15 @@ def reproduce(
         "# {{KERNEL_INVOCATION_PLACEHOLDER}}", invocation_snippet
     )
     out_py_path.write_text(final_code, encoding="utf-8")
-    logger.info(
-        "REPRODUCER_OUTPUT script=%s json=%s kernel=%s",
-        str(out_py_path.resolve()),
-        str(temp_json_path.resolve()),
-        context_bundle.kernel_info.function_name,
-    )
+
+    filepath = context_bundle.kernel_info.file_path
+    filepath = "/".join(filepath.split("/")[5:])
+    ret = {
+        "kernel_src_path": filepath,
+        "kernel": context_bundle.kernel_info.function_name,
+        "repo_script": str(out_py_path.resolve()),
+        "repo_context": str(temp_json_path.resolve()),
+    }
+    logger.info("REPRODUCER_OUTPUT\n%s", ret)
+
+    return ret

{tritonparse-0.2.4.dev20250930071536.dist-info → tritonparse-0.2.4.dev20251001071529.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tritonparse
-Version: 0.2.4.dev20250930071536
+Version: 0.2.4.dev20251001071529
 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.2.4.dev20250930071536.dist-info → tritonparse-0.2.4.dev20251001071529.dist-info}/RECORD

@@ -14,7 +14,7 @@ tritonparse/trace_processor.py,sha256=QzUOKwnOkBbwTTKBsa5ZMUABPLMJIBFtTcG2SkhO0I
 tritonparse/utils.py,sha256=cO3c82PJfToW2pDsVicP3dFh1We3UVv3c3NqC_aTb_g,4312
 tritonparse/reproducer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tritonparse/reproducer/cli.py,sha256=JvnAi1FKSpNa6eHEapRn9jdXxsj1vvyrrEEnfxTJYa8,871
-tritonparse/reproducer/orchestrator.py,sha256=XUq4S44kcFi5rgiOlcjBCEOKqNrJIfHmSdLhIfXzbeo,2441
+tritonparse/reproducer/orchestrator.py,sha256=Uy3CSntjzgd1VZrsHKARE0XTqBgpUukGo7C8b37m2JA,2640
 tritonparse/reproducer/utils.py,sha256=qi4XTKk0pWV4hgYg_GPBISEfVXlrI6tZR0A5ZZbwVyo,11132
 tritonparse/reproducer/ingestion/ndjson.py,sha256=pEujTl5xXW2E2DEW8ngxXQ8qP9oawb90wBVTWHDs1jk,7372
 tritonparse/reproducer/templates/example.py,sha256=XWfXD4tDOiE213YlWWK1l1ZgXbK3BX61NnvuVTkO-S0,8595
@@ -25,8 +25,8 @@ tritonparse/tools/format_fix.py,sha256=Ol0Sjui8D7OzHwbamAfGnq8V5Y63uwNaFTKSORN5H
 tritonparse/tools/load_tensor.py,sha256=tfdmNVd9gsZqO6msQBhbXIhOvUzgc83yF64k2GDWPNk,2122
 tritonparse/tools/prettify_ndjson.py,sha256=r2YlHwFDTHgML7KljRmMsHaDg29q8gOQAgyDKWJhxRM,11062
 tritonparse/tools/readme.md,sha256=w6PWYfYnRgoPArLjxG9rVrpcLUkoVMGuRlbpF-o0IQM,110
-tritonparse-0.2.4.dev20250930071536.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
-tritonparse-0.2.4.dev20250930071536.dist-info/METADATA,sha256=mjc5L2YgsgDRw970IxV0KdFhsWRASAaP8lMiX-pEKXw,6580
-tritonparse-0.2.4.dev20250930071536.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tritonparse-0.2.4.dev20250930071536.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
-tritonparse-0.2.4.dev20250930071536.dist-info/RECORD,,
+tritonparse-0.2.4.dev20251001071529.dist-info/licenses/LICENSE,sha256=4ZciugpyN7wcM4L-9pyDh_etvMUeIfBhDTyH1zeZlQM,1515
+tritonparse-0.2.4.dev20251001071529.dist-info/METADATA,sha256=12KImzhnmkCnXbpSklTxH439eB-0Zs2nxUY04_pklUg,6580
+tritonparse-0.2.4.dev20251001071529.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tritonparse-0.2.4.dev20251001071529.dist-info/top_level.txt,sha256=ITcTKgp3vf_bXV9vixuQU9IrZa3L1EfDSZwvRzRaoJU,12
+tritonparse-0.2.4.dev20251001071529.dist-info/RECORD,,