down into cpp

pytorch · Apr 24, 2024 · 8405149 · 8405149
1 parent 6e3c699
commit 8405149
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/experimental/fori_loop.py b/torch_xla/experimental/fori_loop.py
@@ -118,7 +118,7 @@ def new_body_fn(*carried_inputs):
       # return list(body_fn(*carried_inputs)).extend(additional_inputs)
       res = list(body_fn(*carried_inputs))
       print("res: ", res)
-      trynewres = res[:-1].extend(res[-1])
+      trynewres = res[:-1] + [res[-1]]
       print("trynewres: ", trynewres)
       newres = res[:-1] + list(additional_inputs) + res[-1]
       print("newres: ", newres)