down into cpp

pytorch · Apr 23, 2024 · ea0e663 · ea0e663
1 parent 33fa1fb
commit ea0e663
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_xla/experimental/fori_loop.py b/torch_xla/experimental/fori_loop.py
@@ -63,6 +63,7 @@ def while_loop(cond_fn, body_fn, carried_inputs, additional_inputs=None):
   # carried_inputs: (Tuple of possibly nested dict/list/tuple of tensors)
   if additional_inputs is None:
     additional_inputs = tuple()
+  print("$$$ additional_inputs: ", additional_inputs)
   return _xla_while_loop(cond_fn, body_fn, carried_inputs, additional_inputs)