importXFormNodeRenderer

Author: s-barannikov

llvm/utils/TableGen/GlobalISelEmitter.cpp

@@ -419,6 +419,9 @@ class GlobalISelEmitter final : public GlobalISelMatchTableExecutorEmitter {
   Error importLeafNodeRenderer(RuleMatcher &M, BuildMIAction &MIBuilder,
                                const TreePatternNode &N) const;
 
+  Error importXFormNodeRenderer(RuleMatcher &M, BuildMIAction &MIBuilder,
+                                const TreePatternNode &N) const;
+
   Expected<action_iterator>
   importExplicitUseRenderer(action_iterator InsertPt, RuleMatcher &Rule,
                             BuildMIAction &DstMIBuilder,
@@ -1310,6 +1313,32 @@ Error GlobalISelEmitter::importLeafNodeRenderer(
   return failedImport("unrecognized node " + to_string(N));
 }
 
+// Equivalent of MatcherGen::EmitResultSDNodeXFormAsOperand.
+Error GlobalISelEmitter::importXFormNodeRenderer(
+    RuleMatcher &M, BuildMIAction &MIBuilder, const TreePatternNode &N) const {
+  const Record *XFormRec = N.getOperator();
+  auto I = SDNodeXFormEquivs.find(XFormRec);
+  if (I == SDNodeXFormEquivs.end())
+    return failedImport("SDNodeXForm " + XFormRec->getName() +
+                        " does not have GISel equivalent");
+
+  const Record *XFormEquivRec = I->second;
+  const TreePatternNode &Node = N.getChild(0);
+
+  const Record *XFormOpc = CGP.getSDNodeTransform(XFormRec).first;
+  if (XFormOpc->getName() == "timm") {
+    // If this is a TargetConstant, there won't be a corresponding
+    // instruction to transform. Instead, this will refer directly to an
+    // operand in an instruction's operand list.
+    MIBuilder.addRenderer<CustomOperandRenderer>(*XFormEquivRec,
+                                                 Node.getName());
+  } else {
+    MIBuilder.addRenderer<CustomRenderer>(*XFormEquivRec, Node.getName());
+  }
+
+  return Error::success();
+}
+
 Expected<action_iterator> GlobalISelEmitter::importExplicitUseRenderer(
     action_iterator InsertPt, RuleMatcher &Rule, BuildMIAction &DstMIBuilder,
     const TreePatternNode &Dst) const {
@@ -1326,24 +1355,9 @@ Expected<action_iterator> GlobalISelEmitter::importExplicitUseRenderer(
   }
 
   if (Dst.getOperator()->isSubClassOf("SDNodeXForm")) {
-    auto &Child = Dst.getChild(0);
-    auto I = SDNodeXFormEquivs.find(Dst.getOperator());
-    if (I != SDNodeXFormEquivs.end()) {
-      const Record *XFormOpc = Dst.getOperator()->getValueAsDef("Opcode");
-      if (XFormOpc->getName() == "timm") {
-        // If this is a TargetConstant, there won't be a corresponding
-        // instruction to transform. Instead, this will refer directly to an
-        // operand in an instruction's operand list.
-        DstMIBuilder.addRenderer<CustomOperandRenderer>(*I->second,
-                                                        Child.getName());
-      } else {
-        DstMIBuilder.addRenderer<CustomRenderer>(*I->second, Child.getName());
-      }
-
-      return InsertPt;
-    }
-    return failedImport("SDNodeXForm " + Child.getName() +
-                        " has no custom renderer");
+    if (Error Err = importXFormNodeRenderer(Rule, DstMIBuilder, Dst))
+      return Err;
+    return InsertPt;
   }
 
   if (Dst.getOperator()->isSubClassOf("Instruction")) {