maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

LLT: Add changeNumElements 

This is the element analog of changeElementType/changeElementSize
This commit is contained in:
Matt Arsenault 2020-01-28 13:14:06 -05:00
parent df8f2774b6
commit 24ab761a60
2 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
llvm/include/llvm/Support/LowLevelTypeImpl.h
View File

@ -137,6 +137,12 @@ public:
: LLT::scalar(NewEltSize);
}
/// Return a vector or scalar with the same element type and the new number of
/// elements.
LLT changeNumElements(unsigned NewNumElts) const {
return LLT::scalarOrVector(NewNumElts, getScalarType());
}
bool isByteSized() const { return (getSizeInBits() & 7) == 0; }
unsigned getScalarSizeInBits() const {

23
llvm/unittests/CodeGen/LowLevelTypeTest.cpp
View File

@ -138,6 +138,29 @@ TEST(LowLevelTypeTest, ChangeElementType) {
EXPECT_EQ(V2S32, V2P0.changeElementType(S32));
}
TEST(LowLevelTypeTest, ChangeNumElements) {
const LLT P0 = LLT::pointer(0, 32);
const LLT V2P0 = LLT::vector(2, P0);
const LLT V3P0 = LLT::vector(3, P0);
const LLT S64 = LLT::scalar(64);
const LLT V2S64 = LLT::vector(2, 64);
const LLT V3S64 = LLT::vector(3, 64);
// Vector to scalar
EXPECT_EQ(S64, V2S64.changeNumElements(1));
// Vector to vector
EXPECT_EQ(V3S64, V2S64.changeNumElements(3));
// Scalar to vector
EXPECT_EQ(V2S64, S64.changeNumElements(2));
EXPECT_EQ(P0, V2P0.changeNumElements(1));
EXPECT_EQ(V3P0, V2P0.changeNumElements(3));
EXPECT_EQ(V2P0, P0.changeNumElements(2));
}
#ifdef GTEST_HAS_DEATH_TEST
#ifndef NDEBUG