diff --git a/lib/VMCore/ConstantFold.cpp b/lib/VMCore/ConstantFold.cpp index 8edcb42530f..5fd11c3a2ec 100644 --- a/lib/VMCore/ConstantFold.cpp +++ b/lib/VMCore/ConstantFold.cpp @@ -5,6 +5,7 @@ //===----------------------------------------------------------------------===// #include "llvm/ConstantHandling.h" +#include AnnotationID ConstRules::AID(AnnotationManager::getID("opt::ConstRules", &ConstRules::find)); @@ -51,6 +52,10 @@ class TemplateRules : public ConstRules { const Constant *V2) const { return SubClassName::Div((const ArgType *)V1, (const ArgType *)V2); } + virtual Constant *rem(const Constant *V1, + const Constant *V2) const { + return SubClassName::Rem((const ArgType *)V1, (const ArgType *)V2); + } virtual ConstantBool *lessthan(const Constant *V1, const Constant *V2) const { @@ -114,6 +119,9 @@ class TemplateRules : public ConstRules { inline static Constant *Div(const ArgType *V1, const ArgType *V2) { return 0; } + inline static Constant *Rem(const ArgType *V1, const ArgType *V2) { + return 0; + } inline static ConstantBool *LessThan(const ArgType *V1, const ArgType *V2) { return 0; } @@ -242,11 +250,8 @@ struct PointerRules : public TemplateRules { // different types. This allows the C++ compiler to automatically generate our // constant handling operations in a typesafe and accurate manner. // -template -struct DirectRules - : public TemplateRules > { - +template +struct DirectRules : public TemplateRules { inline static Constant *Add(const ConstantClass *V1, const ConstantClass *V2) { BuiltinType Result = (BuiltinType)V1->getValue() + @@ -268,8 +273,9 @@ struct DirectRules return ConstantClass::get(*Ty, Result); } - inline static Constant *Div(const ConstantClass *V1, + inline static Constant *Div(const ConstantClass *V1, const ConstantClass *V2) { + if (V2->isNullValue()) return 0; BuiltinType Result = (BuiltinType)V1->getValue() / (BuiltinType)V2->getValue(); return ConstantClass::get(*Ty, Result); @@ -317,10 +323,41 @@ struct DirectRules // integer types, but not all types in general. // template -struct DirectIntRules : public DirectRules { +struct DirectIntRules + : public DirectRules > { inline static Constant *Not(const ConstantClass *V) { return ConstantClass::get(*Ty, ~(BuiltinType)V->getValue());; } + + inline static Constant *Rem(const ConstantClass *V1, + const ConstantClass *V2) { + if (V2->isNullValue()) return 0; + BuiltinType Result = (BuiltinType)V1->getValue() % + (BuiltinType)V2->getValue(); + return ConstantClass::get(*Ty, Result); + } +}; + + +//===----------------------------------------------------------------------===// +// DirectFPRules Class +//===----------------------------------------------------------------------===// +// +// DirectFPRules provides implementations of functions that are valid on +// floating point types, but not all types in general. +// +template +struct DirectFPRules + : public DirectRules > { + inline static Constant *Rem(const ConstantClass *V1, + const ConstantClass *V2) { + if (V2->isNullValue()) return 0; + BuiltinType Result = std::fmod((BuiltinType)V1->getValue(), + (BuiltinType)V2->getValue()); + return ConstantClass::get(*Ty, Result); + } }; @@ -360,9 +397,9 @@ Annotation *ConstRules::find(AnnotationID AID, const Annotable *TyA, void *) { case Type::ULongTyID: return new DirectIntRules(); case Type::FloatTyID: - return new DirectRules(); + return new DirectFPRules(); case Type::DoubleTyID: - return new DirectRules(); + return new DirectFPRules(); default: return new EmptyRules(); }