test

pytorch · Apr 10, 2024 · 16f8b5b · 16f8b5b
1 parent 81d3c88
commit 16f8b5b
Showing 1 changed file with 3 additions and 1 deletion.
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,7 +40,9 @@ 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)
+print("expected: ", l_out_expected)
 
 # --- linear two ---
 l_in_2 = torch.randn(10, device=xm.xla_device())