1 //===- TestStackOffset.cpp - StackOffset unit tests------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "llvm/Support/TypeSize.h" 10 #include "AArch64InstrInfo.h" 11 #include "gtest/gtest.h" 12 13 using namespace llvm; 14 15 TEST(StackOffset, decomposeStackOffsetForFrameOffsets) { 16 StackOffset A = StackOffset::getFixed(8); 17 StackOffset B = StackOffset::getFixed(4); 18 StackOffset C = StackOffset::getScalable(16); 19 20 // If all offsets can be materialized with only ADDVL, 21 // make sure PLSized is 0. 22 int64_t ByteSized, VLSized, PLSized; 23 AArch64InstrInfo::decomposeStackOffsetForFrameOffsets(A + B + C, ByteSized, PLSized, 24 VLSized); 25 EXPECT_EQ(12, ByteSized); 26 EXPECT_EQ(1, VLSized); 27 EXPECT_EQ(0, PLSized); 28 29 // If we need an ADDPL to materialize the offset, and the number of scalable 30 // bytes fits the ADDPL immediate, fold the scalable bytes to fit in PLSized. 31 StackOffset D = StackOffset::getScalable(2); 32 AArch64InstrInfo::decomposeStackOffsetForFrameOffsets(C + D, ByteSized, PLSized, VLSized); 33 EXPECT_EQ(0, ByteSized); 34 EXPECT_EQ(0, VLSized); 35 EXPECT_EQ(9, PLSized); 36 37 StackOffset E = StackOffset::getScalable(64); 38 StackOffset F = StackOffset::getScalable(2); 39 AArch64InstrInfo::decomposeStackOffsetForFrameOffsets(E + F, ByteSized, PLSized, VLSized); 40 EXPECT_EQ(0, ByteSized); 41 EXPECT_EQ(0, VLSized); 42 EXPECT_EQ(33, PLSized); 43 44 // If the offset requires an ADDPL instruction to materialize, and would 45 // require more than two instructions, decompose it into both 46 // ADDVL (n x 16 bytes) and ADDPL (n x 2 bytes) instructions. 47 StackOffset G = StackOffset::getScalable(128); 48 StackOffset H = StackOffset::getScalable(2); 49 AArch64InstrInfo::decomposeStackOffsetForFrameOffsets(G + H, ByteSized, PLSized, VLSized); 50 EXPECT_EQ(0, ByteSized); 51 EXPECT_EQ(8, VLSized); 52 EXPECT_EQ(1, PLSized); 53 } 54