test

pytorch · Apr 30, 2024 · 95227f0 · 95227f0
1 parent abafd6a
commit 95227f0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_test_mnist.py b/test/test_test_mnist.py
@@ -348,6 +348,7 @@ def body_fn(upper, lower, one_value, x, input_value, output_value, *args):
     bn_list = []
     bn_flag = False
 
+  print("final additional_inputs: ", additional_inputs)
   # print("in mnist additional_inputs: ", additional_inputs)
   ### linear 10*20 + 20*30
   # upper__, lower__, one_value__, torch_add_res__, input_value__, weight1__, bias1__, w2_, b2_, output_value_real__, = _xla_while_loop(