test

pytorch · Apr 10, 2024 · dcc0103 · dcc0103
1 parent a802df6
commit dcc0103
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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())