54 lines
2.1 KiB
C++
54 lines
2.1 KiB
C++
|
//===- TestStackOffset.cpp - StackOffset unit tests------------------------===//
|
||
|
|
//
|
||
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
||
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
||
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#include "llvm/Support/TypeSize.h"
|
||
|
|
#include "AArch64InstrInfo.h"
|
||
|
|
#include "gtest/gtest.h"
|
||
|
|
|
||
|
|
using namespace llvm;
|
||
|
|
|
||
|
|
TEST(StackOffset, decomposeStackOffsetForFrameOffsets) {
|
||
|
|
StackOffset A = StackOffset::getFixed(8);
|
||
|
|
StackOffset B = StackOffset::getFixed(4);
|
||
|
|
StackOffset C = StackOffset::getScalable(16);
|
||
|
|
|
||
|
|
// If all offsets can be materialized with only ADDVL,
|
||
|
|
// make sure PLSized is 0.
|
||
|
|
int64_t ByteSized, VLSized, PLSized;
|
||
|
|
AArch64InstrInfo::decomposeStackOffsetForFrameOffsets(A + B + C, ByteSized, PLSized,
|
||
|
|
VLSized);
|
||
|
|
EXPECT_EQ(12, ByteSized);
|
||
|
|
EXPECT_EQ(1, VLSized);
|
||
|
|
EXPECT_EQ(0, PLSized);
|
||
|
|
|
||
|
|
// If we need an ADDPL to materialize the offset, and the number of scalable
|
||
|
|
// bytes fits the ADDPL immediate, fold the scalable bytes to fit in PLSized.
|
||
|
|
StackOffset D = StackOffset::getScalable(2);
|
||
|
|
AArch64InstrInfo::decomposeStackOffsetForFrameOffsets(C + D, ByteSized, PLSized, VLSized);
|
||
|
|
EXPECT_EQ(0, ByteSized);
|
||
|
|
EXPECT_EQ(0, VLSized);
|
||
|
|
EXPECT_EQ(9, PLSized);
|
||
|
|
|
||
|
|
StackOffset E = StackOffset::getScalable(64);
|
||
|
|
StackOffset F = StackOffset::getScalable(2);
|
||
|
|
AArch64InstrInfo::decomposeStackOffsetForFrameOffsets(E + F, ByteSized, PLSized, VLSized);
|
||
|
|
EXPECT_EQ(0, ByteSized);
|
||
|
|
EXPECT_EQ(0, VLSized);
|
||
|
|
EXPECT_EQ(33, PLSized);
|
||
|
|
|
||
|
|
// If the offset requires an ADDPL instruction to materialize, and would
|
||
|
|
// require more than two instructions, decompose it into both
|
||
|
|
// ADDVL (n x 16 bytes) and ADDPL (n x 2 bytes) instructions.
|
||
|
|
StackOffset G = StackOffset::getScalable(128);
|
||
|
|
StackOffset H = StackOffset::getScalable(2);
|
||
|
|
AArch64InstrInfo::decomposeStackOffsetForFrameOffsets(G + H, ByteSized, PLSized, VLSized);
|
||
|
|
EXPECT_EQ(0, ByteSized);
|
||
|
|
EXPECT_EQ(8, VLSized);
|
||
|
|
EXPECT_EQ(1, PLSized);
|
||
|
|
}
|