diff options
| -rw-r--r-- | include/llvm/Instruction.h | 3 | ||||
| -rw-r--r-- | lib/VMCore/Instruction.cpp | 6 | ||||
| -rw-r--r-- | unittests/VMCore/IRBuilderTest.cpp | 9 |
3 files changed, 18 insertions, 0 deletions
diff --git a/include/llvm/Instruction.h b/include/llvm/Instruction.h index 2c1d41c705..0f717ad9dd 100644 --- a/include/llvm/Instruction.h +++ b/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/lib/VMCore/Instruction.cpp b/lib/VMCore/Instruction.cpp index d93c1d7a22..7b73e770fb 100644 --- a/lib/VMCore/Instruction.cpp +++ b/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/unittests/VMCore/IRBuilderTest.cpp b/unittests/VMCore/IRBuilderTest.cpp index 8a22b104ba..665cfb3f13 100644 --- a/unittests/VMCore/IRBuilderTest.cpp +++ b/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()); + } } |