Skip to content

Commit 76a9a66

Browse files
author
git apple-llvm automerger
committed
Merge commit '52e50bc1c6c1' from apple/main into swift/next
2 parents 50f7ba4 + 52e50bc commit 76a9a66

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

llvm/unittests/Analysis/TFUtilsTest.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ TEST(TFUtilsTest, LoadAndExecuteTest) {
5757
auto ER = Evaluator.evaluate();
5858
EXPECT_TRUE(ER.hasValue());
5959
float Ret = *ER->getTensorValue<float>(0);
60-
EXPECT_EQ(static_cast<size_t>(Ret), 80);
60+
EXPECT_EQ(static_cast<int64_t>(Ret), 80);
6161
EXPECT_EQ(ER->getUntypedTensorValue(0),
6262
reinterpret_cast<const void *>(ER->getTensorValue<float>(0)));
6363
}
@@ -72,7 +72,7 @@ TEST(TFUtilsTest, LoadAndExecuteTest) {
7272
auto ER = Evaluator.evaluate();
7373
EXPECT_TRUE(ER.hasValue());
7474
float Ret = *ER->getTensorValue<float>(0);
75-
EXPECT_EQ(static_cast<size_t>(Ret), 80);
75+
EXPECT_EQ(static_cast<int64_t>(Ret), 80);
7676
}
7777
}
7878

@@ -135,10 +135,10 @@ TEST(TFUtilsTest, TensorSpecSizesAndTypes) {
135135
auto Spec3DLarge = TensorSpec::createSpec<float>("Hi3", {2, 4, 10});
136136
EXPECT_TRUE(Spec1D.isElementType<int16_t>());
137137
EXPECT_FALSE(Spec3DLarge.isElementType<double>());
138-
EXPECT_EQ(Spec1D.getElementCount(), 1);
139-
EXPECT_EQ(Spec2D.getElementCount(), 1);
140-
EXPECT_EQ(Spec1DLarge.getElementCount(), 10);
141-
EXPECT_EQ(Spec3DLarge.getElementCount(), 80);
138+
EXPECT_EQ(Spec1D.getElementCount(), 1U);
139+
EXPECT_EQ(Spec2D.getElementCount(), 1U);
140+
EXPECT_EQ(Spec1DLarge.getElementCount(), 10U);
141+
EXPECT_EQ(Spec3DLarge.getElementCount(), 80U);
142142
EXPECT_EQ(Spec3DLarge.getElementByteSize(), sizeof(float));
143143
EXPECT_EQ(Spec1D.getElementByteSize(), sizeof(int16_t));
144144
}

0 commit comments

Comments
 (0)