[MSP430] Manually set setCC type for vectors

[INdek]

Summary

Avoids triggering the "No default SetCC type for vectors!" assert.

llvm/lib/Target/MSP430/MSP430ISelLowering.h

public:
     bool isZExtFree(SDValue Val, EVT VT2) const override;
 
     MachineBasicBlock *
     EmitInstrWithCustomInserter(MachineInstr &MI,
                                 MachineBasicBlock *BB) const override;
     MachineBasicBlock *EmitShiftInstr(MachineInstr &MI,
                                       MachineBasicBlock *BB) const;
 
+    EVT getSetCCResultType(const DataLayout &DL, LLVMContext &Context,
+                           EVT VT) const override;
   private:
     SDValue LowerCCCCallTo(SDValue Chain, SDValue Callee,
                            CallingConv::ID CallConv, bool isVarArg,
                            bool isTailCall,
                            const SmallVectorImpl<ISD::OutputArg> &Outs,
                            const SmallVectorImpl<SDValue> &OutVals,
                            const SmallVectorImpl<ISD::InputArg> &Ins,
                            const SDLoc &dl, SelectionDAG &DAG,

llvm/lib/Target/MSP430/MSP430ISelLowering.cpp

BuildMI(*BB, BB->begin(), dl, TII.get(MSP430::PHI), MI.getOperand(0).getReg())
       .addReg(MI.getOperand(2).getReg())
       .addMBB(copy0MBB)
       .addReg(MI.getOperand(1).getReg())
       .addMBB(thisMBB);
 
   MI.eraseFromParent(); // The pseudo instruction is gone now.
   return BB;
 }
+
+EVT MSP430TargetLowering::getSetCCResultType(const DataLayout &DL, LLVMContext &,
+                                             EVT VT) const {
+  if (!VT.isVector())
+    return getPointerTy(DL);
+
+  return VT.changeVectorElementTypeToInteger();
+}

llvm/test/CodeGen/MSP430/get-setcc-result-type.ll

+; RUN: llc -march=msp430 < %s | FileCheck %s
+
+target datalayout = "e-p:16:8:8-i8:8:8-i8:8:8-i32:8:8"
+target triple = "msp430-generic-generic"
+
+define void @getSetCCResultType(<1 x i16>* %p, <1 x i16>* %q) {
+; CHECK-LABEL: getSetCCResultType:
+; CHECK: tst  0(r12)
+; CHECK-NEXT: mov r2, r13
+; CHECK-NEXT: rra r13
+; CHECK-NEXT: and #1, r13
+; CHECK-NEXT: clr r14
+; CHECK-NEXT: sub r13, r14
+; CHECK-NEXT: mov r14, 0(r12)
+; CHECK-NEXT: ret
+entry:
+  %0 = load <1 x i16>, <1 x i16>* %p, align 16
+  %cmp = icmp eq <1 x i16> %0, zeroinitializer
+  %sext = sext <1 x i1> %cmp to <1 x i16>
+  store <1 x i16> %sext, <1 x i16>* %p, align 16
+  ret void
+}