From ea0e663748465fc51f555d196384825cc123dc5b Mon Sep 17 00:00:00 2001 From: manfeibaigithub Date: Tue, 23 Apr 2024 23:36:33 +0000 Subject: [PATCH] down into cpp --- torch_xla/experimental/fori_loop.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torch_xla/experimental/fori_loop.py b/torch_xla/experimental/fori_loop.py index 8ed3a783200..7119947ad2b 100644 --- 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)