diff --git a/docs/TableGenFundamentals.html b/docs/TableGenFundamentals.html index e504a894640..3410cacb62d 100644 --- a/docs/TableGenFundamentals.html +++ b/docs/TableGenFundamentals.html @@ -425,8 +425,8 @@ class. This operation is analogous to $(foreach) in GNU make.
'b' if the result of integer operator 'a' is nonzero, 'c' otherwise.
!eq(a,b)
Integer one if string a is equal to string b, zero otherwise. This - only operates on string objects. Use !cast to compare other - types of objects.
+ only operates on string, int and bit objects. Use !cast to + compare other types of objects.

Note that all of the values have rules specifying how they convert to values diff --git a/test/TableGen/eqbit.td b/test/TableGen/eqbit.td new file mode 100644 index 00000000000..3953252c417 --- /dev/null +++ b/test/TableGen/eqbit.td @@ -0,0 +1,11 @@ +// RUN: tblgen %s | FileCheck %s +// XFAIL: vg_leak +// CHECK: a = 6 +// CHECK: a = 5 + +class A { + int a = !if(!eq(b, 1), 5, 6); +} + +def X : A<0>; +def Y : A; diff --git a/utils/TableGen/Record.cpp b/utils/TableGen/Record.cpp index 4f9f6045c14..d9c5dd30e18 100644 --- a/utils/TableGen/Record.cpp +++ b/utils/TableGen/Record.cpp @@ -721,9 +721,20 @@ Init *BinOpInit::Fold(Record *CurRec, MultiClass *CurMultiClass) { break; } case EQ: { - // Make sure we've resolved + // try to fold eq comparison for 'bit' and 'int', otherwise fallback + // to string objects. + IntInit* L = + dynamic_cast(LHS->convertInitializerTo(new IntRecTy())); + IntInit* R = + dynamic_cast(RHS->convertInitializerTo(new IntRecTy())); + + if (L && R) + return new IntInit(L->getValue() == R->getValue()); + StringInit *LHSs = dynamic_cast(LHS); StringInit *RHSs = dynamic_cast(RHS); + + // Make sure we've resolved if (LHSs && RHSs) return new IntInit(LHSs->getValue() == RHSs->getValue());