test

pytorch · Apr 10, 2024 · c1ca11c · c1ca11c
1 parent bfeeb62
commit c1ca11c
Showing 1 changed file with 5 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
@@ -97,8 +97,12 @@ def body_fun(l_in_i):
 
 fake_l_in_0 = torch.ones([10], dtype=torch.float32, device=device)
 expectation = linear_0(fake_l_in_0)
+print("expectation: ", expectation)
 
-assert l_out_ == expectation
+if (l_out_ == expectation):
+  print("pass")
+else:
+  print("not equal")
 
 # --- linear two ---
 # l_in_2 = torch.randn(10, device=xm.xla_device())