down into cpp

pytorch · Apr 24, 2024 · 7d07234 · 7d07234
1 parent 5b8fcb7
commit 7d07234
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_fori_loop_with_while_loop_simple_add_dispatch_in_torch.py b/test/test_fori_loop_with_while_loop_simple_add_dispatch_in_torch.py
@@ -142,7 +142,7 @@ def test_while_loop_tpu_simple_linear_wrapper(self):
     linear_0 = torch.nn.Linear(10, 20).to(xm.xla_device())
 
     def cond_fn(upper, lower, one_value, x, input_value, output_value):
-      return lower[0] < upper[0]
+      return lower[0] >= upper[0]
 
     def body_fn(upper, lower, one_value, x, input_value, output_value):
       new_lower = torch.add(one_value, lower)