[CodeGen][LLT] Add isFixedVector and isScalableVector (#71713)
The current isScalable function requires a user to call isVector before
hand in order to avoid an assertion failure in the case that the LLT is
not a vector.
This patch addds helper functions that allow a user to query whether the
LLT is fixed or scalable, not wanting an assertion failure in the case
that the LLT was never a vector in the first place.
diff --git a/llvm/unittests/CodeGen/LowLevelTypeTest.cpp b/llvm/unittests/CodeGen/LowLevelTypeTest.cpp
index 7b13fef..8c909c0 100644
--- a/llvm/unittests/CodeGen/LowLevelTypeTest.cpp
+++ b/llvm/unittests/CodeGen/LowLevelTypeTest.cpp
@@ -412,4 +412,18 @@
EXPECT_EQ(LLT::pointer(0, 32), CEP0);
EXPECT_EQ(LLT::scalable_vector(2, 32), CESV2S32);
}
+
+TEST(LowLevelTypeTest, IsFixedVector) {
+ EXPECT_FALSE(LLT::scalar(32).isFixedVector());
+ EXPECT_TRUE(LLT::fixed_vector(2, 32).isFixedVector());
+ EXPECT_FALSE(LLT::scalable_vector(2, 32).isFixedVector());
+ EXPECT_FALSE(LLT::scalable_vector(1, 32).isFixedVector());
+}
+
+TEST(LowLevelTypeTest, IsScalableVector) {
+ EXPECT_FALSE(LLT::scalar(32).isScalableVector());
+ EXPECT_FALSE(LLT::fixed_vector(2, 32).isScalableVector());
+ EXPECT_TRUE(LLT::scalable_vector(2, 32).isScalableVector());
+ EXPECT_TRUE(LLT::scalable_vector(1, 32).isScalableVector());
+}
}
