Skip to content

Commit 2430ddf

Browse files
committed
test fixes
1 parent 9736aae commit 2430ddf

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

test/cpp/test_aten_xla_tensor.cpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -280,9 +280,8 @@ TEST_F(AtenXlaTensorTest, TestSymSizes) {
280280
ForEachDevice([&](const torch::Device& device) {
281281
torch::Tensor a = torch::rand({2, 3}, torch::TensorOptions(torch::kFloat));
282282
torch::Tensor xla_a = CopyToDevice(a, device);
283-
torch::Tensor c = a.sym_sizes(0);
284-
AllClose(a.sym_sizes(0), 2);
285-
AllClose(a.sym_sizes(0).is_symbolic(), false);
283+
ASSERT_EQ(a.sym_sizes().at(0).expect_int(), 2);
284+
ASSERT_EQ(a.sym_sizes().at(0).is_symbolic(), false);
286285
});
287286
}
288287

0 commit comments

Comments
 (0)