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());