From 760d49c4615545fa1aa5669fc7191f93f019be0d Mon Sep 17 00:00:00 2001 From: manfeibaigithub Date: Wed, 24 Apr 2024 22:45:57 +0000 Subject: [PATCH] down into cpp --- torch_xla/experimental/fori_loop.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torch_xla/experimental/fori_loop.py b/torch_xla/experimental/fori_loop.py index 8b5dc7efcc0..3d0fff3d49f 100644 --- a/torch_xla/experimental/fori_loop.py +++ b/torch_xla/experimental/fori_loop.py @@ -116,7 +116,8 @@ def new_body_fn(*carried_inputs): # res.extend(additional_inputs) # print("res: ", res) # return list(body_fn(*carried_inputs)).extend(additional_inputs) - self.named_parameters + # self.named_parameters + weight = self.linear.weight res = list(body_fn(*carried_inputs)) # print("res: ", res) # trynewres = res[:-1] + [res[-1]]