From dcc0103d8696bf9a1592b2a2c0b5a559c368648c Mon Sep 17 00:00:00 2001 From: manfeibaigithub Date: Wed, 10 Apr 2024 19:57:28 +0000 Subject: [PATCH] test --- test/test_fori_loop_simple_linear_model_test_code.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/test_fori_loop_simple_linear_model_test_code.py b/test/test_fori_loop_simple_linear_model_test_code.py index 86ec6d096f80..285a6e0f37e4 100644 --- a/test/test_fori_loop_simple_linear_model_test_code.py +++ b/test/test_fori_loop_simple_linear_model_test_code.py @@ -60,6 +60,8 @@ def body_fun(l_in_i): # TODO(@manfei), need to create new variable to seperate old/formal HLO/IR l_in_0 = torch.randn(10, device=xm.xla_device()) +print("body_fun.weight: ", body_fun.weight) +print("body_fun.weight_: ", body_fun.weight_) # def body_fun(x, y, l_in): # # l_in = torch.randn(10, device=xm.xla_device()) # linear = torch.nn.Linear(10, 20).to(xm.xla_device())