prints for static inputs indices

2025-11-06 09:17:11 +08:00 · 2025-11-04 13:19:14 -08:00
2 changed files with 4 additions and 1 deletions
--- a/torch/_functorch/_aot_autograd/graph_compile.py
+++ b/torch/_functorch/_aot_autograd/graph_compile.py
@ -179,6 +179,9 @@ def aot_stage1_graph_capture(
            )
        )

+    print(f"in aot_stage1_graph_capture. maybe_subclass_meta.fw_metadata.static_input_indices:{maybe_subclass_meta.fw_metadata.static_input_indices if maybe_subclass_meta is not None and maybe_subclass_meta.fw_metadata is not None else None}")
+    print(f"in aot_stage1_graph_capture. aot_state.fw_metadata.static_input_indices:{aot_state.fw_metadata.static_input_indices}")
+
    return AOTGraphCapture(
        wrappers=wrappers,
        graph_module=graph,
--- a/torch/_inductor/compile_fx.py
+++ b/torch/_inductor/compile_fx.py
@ -2318,7 +2318,7 @@ def compile_fx_forward(
    # force the outputs of invoke_subgraph subgraph to follow the
    # original strides
    _recursive_record_user_visible_output_idxs(gm)
-
+    print(f"in compile_fx_foward. static_input_idxs:{get_static_input_idxs(fixed)}")
    return inner_compile(
        gm,
        example_inputs,