From 16f8b5ba665d52c5d938936352f5757eac73b8d5 Mon Sep 17 00:00:00 2001 From: manfeibaigithub Date: Wed, 10 Apr 2024 23:54:11 +0000 Subject: [PATCH] test --- test/test_fori_loop_simple_linear_model_test_code.py | 4 +++- 1 file changed, 3 insertions(+), 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 index 9866fc90526..4c0f96526fe 100644 --- 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())