test

pytorch · Apr 10, 2024 · b86ec8b · b86ec8b
1 parent 16f8b5b
commit b86ec8b
Showing 1 changed file with 5 additions and 2 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
@@ -40,10 +40,13 @@ def body_fun(l_in_i):
 print("l_in_i_plus_1_: ", l_in_i_plus_1_)
 print("l_out_: ", l_out_)
 
-for i in range(5):
-  l_out_expected = linear_0(l_in_0)
+l_out_expected = linear_0(l_in_0)
 print("expected: ", l_out_expected)
 
+# for i in range(5):
+#   l_out_expected = linear_0(l_in_0)
+# print("expected: ", l_out_expected)
+
 # --- linear two ---
 l_in_2 = torch.randn(10, device=xm.xla_device())
 linear_2 = torch.nn.Linear(10, 20).to(xm.xla_device())