diff options
| -rw-r--r-- | llvm/include/llvm/Instruction.h | 3 | ||||
| -rw-r--r-- | llvm/lib/VMCore/Instruction.cpp | 6 | ||||
| -rw-r--r-- | llvm/unittests/VMCore/IRBuilderTest.cpp | 9 |
3 files changed, 18 insertions, 0 deletions
diff --git a/llvm/include/llvm/Instruction.h b/llvm/include/llvm/Instruction.h index 2c1d41c705b..0f717ad9ddd 100644 --- a/llvm/include/llvm/Instruction.h +++ b/llvm/include/llvm/Instruction.h @@ -227,6 +227,9 @@ public: /// these flats. FastMathFlags getFastMathFlags() const; + /// Copy I's fast-math flags + void copyFastMathFlags(const Instruction *I); + private: /// hasMetadataHashEntry - Return true if we have an entry in the on-the-side /// metadata hash. diff --git a/llvm/lib/VMCore/Instruction.cpp b/llvm/lib/VMCore/Instruction.cpp index d93c1d7a22c..7b73e770fb3 100644 --- a/llvm/lib/VMCore/Instruction.cpp +++ b/llvm/lib/VMCore/Instruction.cpp @@ -177,6 +177,12 @@ FastMathFlags Instruction::getFastMathFlags() const { return cast<FPMathOperator>(this)->getFastMathFlags(); } +/// Copy I's fast-math flags +void Instruction::copyFastMathFlags(const Instruction *I) { + setFastMathFlags(I->getFastMathFlags()); +} + + const char *Instruction::getOpcodeName(unsigned OpCode) { switch (OpCode) { // Terminators diff --git a/llvm/unittests/VMCore/IRBuilderTest.cpp b/llvm/unittests/VMCore/IRBuilderTest.cpp index 8a22b104ba6..665cfb3f138 100644 --- a/llvm/unittests/VMCore/IRBuilderTest.cpp +++ b/llvm/unittests/VMCore/IRBuilderTest.cpp @@ -164,6 +164,15 @@ TEST_F(IRBuilderTest, FastMathFlags) { FDiv = cast<Instruction>(F); EXPECT_TRUE(FDiv->hasAllowReciprocal()); + Builder.clearFastMathFlags(); + + F = Builder.CreateFDiv(F, F); + ASSERT_TRUE(isa<Instruction>(F)); + FDiv = cast<Instruction>(F); + EXPECT_FALSE(FDiv->getFastMathFlags().any()); + FDiv->copyFastMathFlags(FAdd); + EXPECT_TRUE(FDiv->hasNoNaNs()); + } } |
