blob: 8d888d8c50f4af2eee52602d8b82ec29a8326645 [file] [edit]
//===- KnownFPClassTest.cpp -----------------------------------------------===//
//
// 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 "GISelMITest.h"
#include "llvm/ADT/FloatingPointMode.h"
#include "llvm/CodeGen/GlobalISel/GISelValueTracking.h"
#include "llvm/CodeGen/GlobalISel/MachineIRBuilder.h"
#include "gtest/gtest.h"
#include <optional>
TEST_F(AArch64GISelMITest, TestFPClassCstPosZero) {
StringRef MIRString = " %3:_(s32) = G_FCONSTANT float 0.0\n"
" %4:_(s32) = COPY %3\n";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
unsigned CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosZero, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCstNegZero) {
StringRef MIRString = " %3:_(s32) = G_FCONSTANT float -0.0\n"
" %4:_(s32) = COPY %3\n";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegZero, Known.KnownFPClasses);
EXPECT_EQ(true, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassUndef) {
StringRef MIRString = R"(
%def:_(s32) = G_IMPLICIT_DEF
%copy_def:_(s32) = COPY %def
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCstVecNegZero) {
StringRef MIRString = R"(
%c0:_(s32) = G_FCONSTANT float -0.0
%c1:_(s32) = G_FCONSTANT float -0.0
%c2:_(s32) = G_FCONSTANT float -0.0
%vector:_(<3 x s32>) = G_BUILD_VECTOR %c0, %c1, %c2
%copy_vector:_(<3 x s32>) = COPY %vector
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegZero, Known.KnownFPClasses);
EXPECT_EQ(true, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCstZeroFPExt) {
StringRef MIRString = R"(
%c0:_(s32) = G_FCONSTANT float 0.0
%ext:_(s64) = nnan ninf G_FPEXT %c0
%copy_vector:_(s64) = COPY %ext
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosZero, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCstVecZeroFPExt) {
StringRef MIRString = R"(
%c0:_(s32) = G_FCONSTANT float 0.0
%c1:_(s32) = G_FCONSTANT float 0.0
%c2:_(s32) = G_FCONSTANT float 0.0
%vector:_(<3 x s32>) = G_BUILD_VECTOR %c0, %c1, %c2
%ext:_(<3 x s64>) = nnan ninf G_FPEXT %vector
%copy_vector:_(<3 x s64>) = COPY %ext
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosZero, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCstZeroFPTrunc) {
StringRef MIRString = R"(
%c0:_(s64) = G_FCONSTANT double 0.0
%trunc:_(s32) = nnan ninf G_FPTRUNC %c0
%copy_vector:_(s32) = COPY %trunc
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite | fcNegZero, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCstVecZeroFPTrunc) {
StringRef MIRString = R"(
%c0:_(s64) = G_FCONSTANT double 0.0
%c1:_(s64) = G_FCONSTANT double 0.0
%c2:_(s64) = G_FCONSTANT double 0.0
%vector:_(<3 x s64>) = G_BUILD_VECTOR %c0, %c1, %c2
%trunc:_(<3 x s32>) = nnan ninf G_FPTRUNC %vector
%copy_vector:_(<3 x s32>) = COPY %trunc
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite | fcNegZero, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectPos0) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_FCONSTANT float 0.0
%rhs:_(s32) = G_FCONSTANT float 0.0
%sel:_(s32) = G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosZero, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectNeg0) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_FCONSTANT float -0.0
%rhs:_(s32) = G_FCONSTANT float -0.0
%sel:_(s32) = G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegZero, Known.KnownFPClasses);
EXPECT_EQ(true, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectPosOrNeg0) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_FCONSTANT float -0.0
%rhs:_(s32) = G_FCONSTANT float 0.0
%sel:_(s32) = G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcZero, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectPosInf) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_FCONSTANT float 0x7FF0000000000000
%rhs:_(s32) = G_FCONSTANT float 0x7FF0000000000000
%sel:_(s32) = G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosInf, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectNegInf) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_FCONSTANT float 0xFFF0000000000000
%rhs:_(s32) = G_FCONSTANT float 0xFFF0000000000000
%sel:_(s32) = G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegInf, Known.KnownFPClasses);
EXPECT_EQ(true, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectPosOrNegInf) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_FCONSTANT float 0x7FF0000000000000
%rhs:_(s32) = G_FCONSTANT float 0xFFF0000000000000
%sel:_(s32) = G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectNNaN) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%sel:_(s32) = nnan G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(~fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectNInf) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%sel:_(s32) = ninf G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(~fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectNNaNNInf) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%cond:_(s1) = G_LOAD %ptr(p0) :: (load (s1))
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%sel:_(s32) = nnan ninf G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(~(fcNan | fcInf), Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFNegNInf) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fneg:_(s32) = ninf G_FNEG %val
%copy_fneg:_(s32) = COPY %fneg
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(~fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFabsUnknown) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = G_FABS %val
%copy_fabs:_(s32) = COPY %fabs
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive | fcNan, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassVecFabsUnknown) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(<3 x s32>) = G_LOAD %ptr(p0) :: (load (<3 x s32>))
%fabs:_(<3 x s32>) = G_FABS %val
%copy_fabs:_(<3 x s32>) = COPY %fabs
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive | fcNan, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFnegFabs) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = G_FABS %val
%fneg:_(s32) = G_FNEG %fabs
%copy_fneg:_(s32) = COPY %fneg
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegative | fcNan, Known.KnownFPClasses);
EXPECT_EQ(true, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFnegFabsNInf) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = ninf G_FABS %val
%fneg:_(s32) = G_FNEG %fabs
%copy_fneg:_(s32) = COPY %fneg
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegFinite | fcNan, Known.KnownFPClasses);
EXPECT_EQ(true, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFnegFabsNNan) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan G_FABS %val
%fneg:_(s32) = G_FNEG %fabs
%copy_fneg:_(s32) = COPY %fneg
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegative, Known.KnownFPClasses);
EXPECT_EQ(true, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCopySignNNanSrc0) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%mag:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%sgn:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan G_FABS %mag
%fcopysign:_(s32) = G_FCOPYSIGN %fabs, %sgn
%copy_fcopysign:_(s32) = COPY %fcopysign
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(~fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCopySignNInfSrc0_NegSign) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%mag:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%sgn:_(s32) = G_FCONSTANT float -1.0
%fabs:_(s32) = ninf G_FLOG %mag
%fcopysign:_(s32) = G_FCOPYSIGN %fabs, %sgn
%copy_fcopysign:_(s32) = COPY %fcopysign
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNan | fcNegZero | fcNegNormal, Known.KnownFPClasses);
EXPECT_EQ(true, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCopySignNInfSrc0_PosSign) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%mag:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%sgn:_(s32) = G_FCONSTANT float 1.0
%fabs:_(s32) = ninf G_FSQRT %mag
%fcopysign:_(s32) = G_FCOPYSIGN %fabs, %sgn
%copy_fcopysign:_(s32) = COPY %fcopysign
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNan | fcPosZero | fcPosNormal, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassUIToFP) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%cast:_(s32) = G_UITOFP %val
%copy_cast:_(s32) = COPY %cast
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosZero | fcPosNormal, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSIToFP) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%cast:_(s32) = G_SITOFP %val
%copy_cast:_(s32) = COPY %cast
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosNormal | fcNegNormal | fcPosZero, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAdd) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fadd:_(s32) = G_FADD %lhs, %rhs
%copy_fadd:_(s32) = COPY %fadd
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAdd_Zero) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_FCONSTANT float 0.0
%fadd:_(s32) = G_FADD %lhs, %rhs
%copy_fadd:_(s32) = COPY %fadd
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags & ~fcNegZero, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAdd_NegZero) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_FCONSTANT float -0.0
%fadd:_(s32) = G_FADD %lhs, %rhs
%copy_fadd:_(s32) = COPY %fadd
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFstrictAdd_Zero) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_FCONSTANT float 0.0
%fadd:_(s32) = G_STRICT_FADD %lhs, %rhs
%copy_fadd:_(s32) = COPY %fadd
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags & ~fcNegZero, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFMul) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%load:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%val:_(s32) = G_FREEZE %load
%fmul:_(s32) = G_FMUL %val, %val
%copy_fadd:_(s32) = COPY %fmul
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive | fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFMulZero) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_FCONSTANT float 0.0
%fabs:_(s32) = nnan ninf G_FABS %lhs
%fmul:_(s32) = G_FMUL %fabs, %rhs
%copy_fadd:_(s32) = COPY %fmul
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosZero, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFLogNeg) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan ninf G_FABS %val
%fneg:_(s32) = nnan ninf G_FNEG %fabs
%flog:_(s32) = G_FLOG %fneg
%copy_flog:_(s32) = COPY %flog
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNan | fcNegInf | fcPosZero | fcNormal, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFLogPosZero) {
StringRef MIRString = R"(
%val:_(s32) = G_FCONSTANT float 0.0
%flog:_(s32) = G_FLOG %val
%copy_flog:_(s32) = COPY %flog
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegInf | fcPosZero | fcNormal, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFLogNegZero) {
StringRef MIRString = R"(
%val:_(s32) = G_FCONSTANT float -0.0
%flog:_(s32) = G_FLOG %val
%copy_flog:_(s32) = COPY %flog
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNegInf | fcPosZero | fcNormal, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassCopy) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = G_FABS %val
%copy:_(s32) = COPY %fabs
%copy_copy:_(s32) = COPY %copy
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(~fcNegative, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassSelectIsFPClass) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%lhs:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%rhs:_(s32) = G_FCONSTANT float 0.0
%cond:_(s1) = G_IS_FPCLASS %lhs, 96
%sel:_(s32) = G_SELECT %cond, %lhs, %rhs
%copy_sel:_(s32) = COPY %sel
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcZero, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFLDExp) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%exp:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = G_FABS %val
%fldexp:_(s32) = G_FLDEXP %fabs, %exp
%copy_fldexp:_(s32) = COPY %fldexp
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive | fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFPowIEvenExp) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%pwr:_(s32) = G_CONSTANT i32 2
%fpowi:_(s32) = G_FPOWI %val, %pwr
%copy_fpowi:_(s32) = COPY %fpowi
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive | fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFPowIPos) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%pwr:_(s32) = G_CONSTANT i32 3
%fabs:_(s32) = nnan ninf G_FABS %val
%fpowi:_(s32) = G_FPOWI %fabs, %pwr
%copy_fpowi:_(s32) = COPY %fpowi
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFPowIInf) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%load:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%val:_(s32) = G_FREEZE %load
%x:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%finite:_(s32) = ninf G_FNEG %val
%normal:_(s32) = G_FCONSTANT float 2.0
%zero_or_nan:_(s32) = G_FREM %val, %val
%nonneg_mask:_(s32) = G_CONSTANT i32 2147483647
%nonneg:_(s32) = G_AND %x, %nonneg_mask
%one:_(s32) = G_CONSTANT i32 1
%two:_(s32) = G_CONSTANT i32 2
%negone:_(s32) = G_CONSTANT i32 -1
%fpowi0:_(s32) = G_FPOWI %finite, %one
%copy_fpowi0:_(s32) = COPY %fpowi0
%fpowi1:_(s32) = G_FPOWI %finite, %two
%copy_fpowi1:_(s32) = COPY %fpowi1
%fpowi2:_(s32) = G_FPOWI %normal, %negone
%copy_fpowi2:_(s32) = COPY %fpowi2
%fpowi3:_(s32) = G_FPOWI %zero_or_nan, %nonneg
%copy_fpowi3:_(s32) = COPY %fpowi3
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
GISelValueTracking Info(*MF);
// powi(finite, 1) --> ~fcInf
Register CopyReg0 = Copies[Copies.size() - 4];
MachineInstr *FinalCopy0 = MRI->getVRegDef(CopyReg0);
Register SrcReg0 = FinalCopy0->getOperand(1).getReg();
KnownFPClass Known0 = Info.computeKnownFPClass(SrcReg0, fcAllFlags);
KnownFPClass KnownInf0 = Info.computeKnownFPClass(SrcReg0, fcInf);
EXPECT_EQ(~fcInf, Known0.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known0.SignBit);
EXPECT_TRUE(KnownInf0.isKnownNeverInfinity());
// powi(finite, 2) --> fcPositive | fcNan
Register CopyReg1 = Copies[Copies.size() - 3];
MachineInstr *FinalCopy1 = MRI->getVRegDef(CopyReg1);
Register SrcReg1 = FinalCopy1->getOperand(1).getReg();
KnownFPClass Known1 = Info.computeKnownFPClass(SrcReg1, fcAllFlags);
KnownFPClass KnownInf1 = Info.computeKnownFPClass(SrcReg1, fcInf);
EXPECT_EQ(fcPositive | fcNan, Known1.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known1.SignBit);
EXPECT_FALSE(KnownInf1.isKnownNeverInfinity());
// powi(normal, -1) --> fcPosFinite
Register CopyReg2 = Copies[Copies.size() - 2];
MachineInstr *FinalCopy2 = MRI->getVRegDef(CopyReg2);
Register SrcReg2 = FinalCopy2->getOperand(1).getReg();
KnownFPClass Known2 = Info.computeKnownFPClass(SrcReg2, fcAllFlags);
KnownFPClass KnownInf2 = Info.computeKnownFPClass(SrcReg2, fcInf);
EXPECT_EQ(fcPosFinite, Known2.KnownFPClasses);
EXPECT_EQ(false, Known2.SignBit);
EXPECT_TRUE(KnownInf2.isKnownNeverInfinity());
// powi(zero_or_nan, nonneg) --> ~fcInf
Register CopyReg3 = Copies[Copies.size() - 1];
MachineInstr *FinalCopy3 = MRI->getVRegDef(CopyReg3);
Register SrcReg3 = FinalCopy3->getOperand(1).getReg();
KnownFPClass KnownInf3 = Info.computeKnownFPClass(SrcReg3, fcInf);
EXPECT_TRUE(KnownInf3.isKnownNeverInfinity());
// TODO: Add powi(0/nan, exp), exp > 0 --> fcNan | fcZero | fcPosNormal
}
TEST_F(AArch64GISelMITest, TestFPClassFDiv) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%load:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%val:_(s32) = G_FREEZE %load
%fdiv:_(s32) = G_FDIV %val, %val
%copy_fdiv:_(s32) = COPY %fdiv
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosNormal | fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFDiv_Inf) {
StringRef MIRString = R"(
%lhs:_(s32) = G_FCONSTANT float 1.0
%rhs:_(s32) = G_FCONSTANT float 0.0
%fdiv:_(s32) = G_FDIV %lhs, %rhs
%copy_fdiv:_(s32) = COPY %fdiv
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosInf, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFRem) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%load:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%val:_(s32) = G_FREEZE %load
%frem:_(s32) = G_FREM %val, %val
%copy_frem:_(s32) = COPY %frem
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcZero | fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFRemSelf_KnownFiniteNonZero) {
// X % X where X is a known-finite, known-nonzero value should produce
// exactly [+-]0.0 (no NaN possible).
StringRef MIRString = R"(
%val:_(s32) = G_FCONSTANT float 2.0
%frem:_(s32) = G_FREM %val, %val
%copy_frem:_(s32) = COPY %frem
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
// 2.0 % 2.0 = 0.0 exactly — NaN is impossible since 2.0 is finite and
// nonzero.
EXPECT_EQ(fcZero, Known.KnownFPClasses);
}
TEST_F(AArch64GISelMITest, TestFPClassShuffleVec) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%vec:_(<4 x s32>) = G_LOAD %ptr(p0) :: (load (<4 x s32>))
%fabs:_(<4 x s32>) = nnan ninf G_FABS %vec
%def:_(<4 x s32>) = G_IMPLICIT_DEF
%shuf:_(<4 x s32>) = G_SHUFFLE_VECTOR %fabs(<4 x s32>), %def, shufflemask(0, 0, 0, 0)
%copy_shuf:_(<4 x s32>) = COPY %shuf
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassBuildVec) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val1:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan ninf G_FABS %val1
%val2:_(s32) = G_FCONSTANT float 3.0
%vec:_(<2 x s32>) = G_BUILD_VECTOR %fabs, %val2
%copy_vec:_(<2 x s32>) = COPY %vec
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassConcatVec) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%vec1:_(<2 x s32>) = G_LOAD %ptr(p0) :: (load (<2 x s32>))
%c1:_(s32) = G_FCONSTANT float 1.0
%c2:_(s32) = G_FCONSTANT float 2.0
%vec2:_(<2 x s32>) = G_BUILD_VECTOR %c1, %c2
%fabs1:_(<2 x s32>) = nnan ninf G_FABS %vec1
%fabs2:_(<2 x s32>) = nnan ninf G_FABS %vec2
%cat:_(<4 x s32>) = G_CONCAT_VECTORS %fabs1, %fabs2
%copy_cat:_(<4 x s32>) = COPY %cat
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassVecExtractElem) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%vec:_(<4 x s32>) = G_LOAD %ptr(p0) :: (load (<4 x s32>))
%fabs:_(<4 x s32>) = nnan ninf G_FABS %vec
%idx:_(s64) = G_CONSTANT i64 1
%extract:_(s32) = G_EXTRACT_VECTOR_ELT %fabs, %idx
%copy_elem:_(s32) = COPY %extract
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassVecInsertElem) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%vec:_(<4 x s32>) = G_LOAD %ptr(p0) :: (load (<4 x s32>))
%fabs1:_(<4 x s32>) = nnan ninf G_FABS %vec
%elem:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs2:_(s32) = nnan ninf G_FABS %elem
%idx:_(s64) = G_CONSTANT i64 1
%insert:_(<4 x s32>) = G_INSERT_VECTOR_ELT %fabs1, %fabs2, %idx
%copy_insert:_(<4 x s32>) = COPY %insert
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFSinh) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fsinh:_(s32) = G_FSINH %val
%copy:_(s32) = COPY %fsinh
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFSinhPos) {
// sinh is sign-preserving: non-negative input → non-negative output.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan ninf G_FABS %val
%fsinh:_(s32) = G_FSINH %fabs
%copy:_(s32) = COPY %fsinh
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFCosh) {
// cosh(x) >= 1 for all real x; never negative.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fcosh:_(s32) = G_FCOSH %val
%copy:_(s32) = COPY %fcosh
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive | fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFCoshNNaN) {
// cosh of a non-NaN source is non-NaN (and non-negative).
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan G_FABS %val
%fcosh:_(s32) = G_FCOSH %fabs
%copy:_(s32) = COPY %fcosh
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPositive, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFTanh) {
// tanh is bounded to (-1, 1): never Inf.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%ftanh:_(s32) = G_FTANH %val
%copy:_(s32) = COPY %ftanh
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags & ~fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFTanhPos) {
// tanh is sign-preserving and bounded to (-1,1): non-negative finite output.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan ninf G_FABS %val
%ftanh:_(s32) = G_FTANH %fabs
%copy:_(s32) = COPY %ftanh
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAsin) {
// asin is bounded to [-π/2, π/2]: never Inf.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fasin:_(s32) = G_FASIN %val
%copy:_(s32) = COPY %fasin
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags & ~fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAsinPos) {
// asin is sign-preserving and bounded: non-negative finite output.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan ninf G_FABS %val
%fasin:_(s32) = G_FASIN %fabs
%copy:_(s32) = COPY %fasin
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAcos) {
// acos is bounded to [0, π]: never Inf, never negative.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%facos:_(s32) = G_FACOS %val
%copy:_(s32) = COPY %facos
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite | fcNan, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAtan) {
// atan is bounded to (-π/2, π/2): never Inf (atan(±Inf) = ±π/2, finite).
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fatan:_(s32) = G_FATAN %val
%copy:_(s32) = COPY %fatan
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags & ~fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAtanPos) {
// atan is sign-preserving and bounded: non-negative finite output.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan ninf G_FABS %val
%fatan:_(s32) = G_FATAN %fabs
%copy:_(s32) = COPY %fatan
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcPosFinite, Known.KnownFPClasses);
EXPECT_EQ(false, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFTan) {
// tan(±Inf) = NaN, tan(finite) = finite: never Inf.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%ftan:_(s32) = G_FTAN %val
%copy:_(s32) = COPY %ftan
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags & ~fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFTanNNaN) {
// tan of a non-NaN, non-Inf source is non-NaN and non-Inf.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs:_(s32) = nnan ninf G_FABS %val
%ftan:_(s32) = G_FTAN %fabs
%copy:_(s32) = COPY %ftan
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcFinite, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAtan2) {
// atan2 result is in (-π, π]: never Inf.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%y:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%x:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fatan2:_(s32) = G_FATAN2 %y, %x
%copy:_(s32) = COPY %fatan2
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcAllFlags & ~fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
TEST_F(AArch64GISelMITest, TestFPClassFAtan2NNaN) {
// atan2 with two non-NaN inputs is non-NaN and non-Inf.
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%y:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%x:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%fabs_y:_(s32) = nnan ninf G_FABS %y
%fabs_x:_(s32) = nnan ninf G_FABS %x
%fatan2:_(s32) = G_FATAN2 %fabs_y, %fabs_x
%copy:_(s32) = COPY %fatan2
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcFinite, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
// isAbsoluteValueULEOne: x - floor(x) is in [0, 1), so multiplying a known-
// finite value by it cannot overflow to infinity.
TEST_F(AArch64GISelMITest, TestFPClassFMulAbsULEOne) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%x:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%val:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%floor:_(s32) = G_FFLOOR %x
%fract:_(s32) = G_FSUB %x, %floor
%finite:_(s32) = nnan ninf G_FABS %val
%fmul:_(s32) = G_FMUL %finite, %fract
%copy:_(s32) = COPY %fmul
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(~fcInf, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}
// G_FMA with A == B (and A guaranteed not-undef): the multiply part is a
// square, so the result is known non-negative (never fcNegative).
TEST_F(AArch64GISelMITest, TestFPClassFMASelfSquare) {
StringRef MIRString = R"(
%ptr:_(p0) = G_IMPLICIT_DEF
%load:_(s32) = G_LOAD %ptr(p0) :: (load (s32))
%val:_(s32) = G_FREEZE %load
%c:_(s32) = G_FCONSTANT float 1.0
%fma:_(s32) = G_FMA %val, %val, %c
%copy:_(s32) = COPY %fma
)";
setUp(MIRString);
if (!TM)
GTEST_SKIP();
Register CopyReg = Copies[Copies.size() - 1];
MachineInstr *FinalCopy = MRI->getVRegDef(CopyReg);
Register SrcReg = FinalCopy->getOperand(1).getReg();
GISelValueTracking Info(*MF);
KnownFPClass Known = Info.computeKnownFPClass(SrcReg);
EXPECT_EQ(fcNan | fcPosInf | fcPosNormal, Known.KnownFPClasses);
EXPECT_EQ(std::nullopt, Known.SignBit);
}