From a197c8accfc5f3f21f11f5e45c734da6b7709fb1 Mon Sep 17 00:00:00 2001 From: manfeibaigithub Date: Thu, 21 Mar 2024 18:43:28 +0000 Subject: [PATCH] test range --- ...st_fori_loop_with_while_loop_simple_add_dispatch_in_torch.py | 2 +- 1 file changed, 1 insertion(+), 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 index 53f6abf8547..0f01d170e87 100644 --- 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 @@ -81,7 +81,7 @@ def test_fori_loop_tpu_addition(self): upper = torch.tensor([30], dtype=torch.int32, device=device) one_value = torch.tensor([1], dtype=torch.int32, device=device) init_val_list = (init_val, one_value) - lowers = torch.tensor([1, 1, 1], dtype=torch.int32, device=device) # lower, init_val, one_value + lowers = torch.tensor([[1], [1], [1]], dtype=torch.int32, device=device) # lower, init_val, one_value def body_fun(a, b): return torch.add(a, b) # [0])