down into cpp

pytorch · Apr 24, 2024 · 760d49c · 760d49c
1 parent 3aa67e7
commit 760d49c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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]]