update

pytorch · Apr 16, 2024 · a79a609 · a79a609
1 parent 2e33759
commit a79a609
Showing 1 changed file with 2 additions and 2 deletions.
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
@@ -94,8 +94,8 @@ def test_while_loop_tpu_simple_linear(self):
 
     # def forward(self, upper, lower, one_value, x, input_value, output_value):
     linear_0 = torch.nn.Linear(10, 20).to(xm.xla_device())
-    weight_0 = linear_0.weight_
-    bias_0 = linear_0.bias_
+    weight_0 = linear_0.weight
+    bias_0 = linear_0.bias
 
     def cond_fn(upper, lower, one_value, x, input_value, weight_0, bias_0, output_value):
       return lower[0] < upper[0]