DAG: propagate ConsecutiveRegs flags to returns too.

[t.p.northover]

Summary

Arguments already have a flag to inform backends when they have been split up. The AArch64 arm64_32 ABI makes use of these on return types too, so that code emitted for armv7k can be ABI-compliant.

There should be no CodeGen changes yet, just making more information available.

[arsenm]

llvm/lib/CodeGen/SelectionDAG/SelectionDAGBuilder.cpp

1643

The hook name is inaccurate here. Maybe there should be an alias for it for the return value?

[t.p.northover]

llvm/lib/CodeGen/SelectionDAG/SelectionDAGBuilder.cpp

1643

I don't think it merits two separate callbacks, returns and parameters are basically the same thing and I'd rather bend the meaning of "argument" if it comes down to it. It's not even really a change to existing practice: OutputArg and ArgFlagsTy are already used in visitRet.

I did put the code actually using this variable into the wrong patch though. I'll upload a new version to fix that and add context while we muse on this.

[t.p.northover]

Added context, and moved use of NeedsRegBlock from incorrect patch I'd uploaded elsewhere.

[t.p.northover]

Ping.

[arsenm]

LGTM

[t.p.northover]

Thanks, *this* one is r358399.

llvm/lib/CodeGen/SelectionDAG/SelectionDAGBuilder.cpp

 
       const Function *F = I.getParent()->getParent();
 
+      bool NeedsRegBlock = TLI.functionArgumentNeedsConsecutiveRegisters(
+          I.getOperand(0)->getType(), F->getCallingConv(),
+          /*IsVarArg*/ false);
+
       ISD::NodeType ExtendKind = ISD::ANY_EXTEND;
       if (F->getAttributes().hasAttribute(AttributeList::ReturnIndex,
                                           Attribute::SExt))

     // points into the callers stack frame.
     CLI.IsTailCall = false;
   } else {
+    bool NeedsRegBlock = functionArgumentNeedsConsecutiveRegisters(
+        CLI.RetTy, CLI.CallConv, CLI.IsVarArg);
     for (unsigned I = 0, E = RetTys.size(); I != E; ++I) {
+      ISD::ArgFlagsTy Flags;
+      if (NeedsRegBlock) {
+        Flags.setInConsecutiveRegs();
+        if (I == RetTys.size() - 1)
+          Flags.setInConsecutiveRegsLast();
+      }
       EVT VT = RetTys[I];
       MVT RegisterVT = getRegisterTypeForCallingConv(CLI.RetTy->getContext(),
                                                      CLI.CallConv, VT);