Add missing aot_autograd_arg_pos_to_source

[ghstack-poisoned]
pytorch · Mar 23, 2023 · 4f198e8 · 4f198e8
1 parent 12da0c7
commit 4f198e8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch/_functorch/aot_autograd.py b/torch/_functorch/aot_autograd.py
@@ -2491,6 +2491,7 @@ def call_compiled_backward():
                                 aot_config.keep_inference_input_mutations,
                                 aot_config.dynamic_shapes,
                                 inference_compiler=None,
+                                aot_autograd_arg_pos_to_source=None,
                             )
                         )
                     else:
@@ -2805,6 +2806,7 @@ def aot_function(
         aot_id=next(AOT_COUNTER),
         keep_inference_input_mutations=keep_inference_input_mutations,
         dynamic_shapes=dynamic,
+        aot_autograd_arg_pos_to_source=None,
     )
     cached_res = None