Skip to content

Commit d2e495c

Browse files
committed
[RISCV][TTI] Add llvm.vp.select into canSplatOperand.
1 parent a30fd03 commit d2e495c

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed

llvm/lib/Target/RISCV/RISCVTargetTransformInfo.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2373,6 +2373,7 @@ bool RISCVTTIImpl::canSplatOperand(Instruction *I, int Operand) const {
23732373
case Intrinsic::vp_ssub_sat:
23742374
case Intrinsic::usub_sat:
23752375
case Intrinsic::vp_usub_sat:
2376+
case Intrinsic::vp_select:
23762377
return Operand == 1;
23772378
// These intrinsics are commutative.
23782379
case Intrinsic::vp_add:

llvm/test/CodeGen/RISCV/rvv/sink-splat-operands.ll

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -5502,21 +5502,20 @@ for.cond.cleanup: ; preds = %vector.body
55025502
define void @sink_splat_vp_select_op1(ptr %a, i32 %x, i32 %vl) {
55035503
; CHECK-LABEL: sink_splat_vp_select_op1:
55045504
; CHECK: # %bb.0: # %entry
5505+
; CHECK-NEXT: lui a4, 1
5506+
; CHECK-NEXT: li a3, 42
5507+
; CHECK-NEXT: slli a5, a2, 32
5508+
; CHECK-NEXT: add a2, a0, a4
5509+
; CHECK-NEXT: srli a4, a5, 32
55055510
; CHECK-NEXT: vsetivli zero, 4, e32, m1, ta, ma
5506-
; CHECK-NEXT: vmv.v.x v8, a1
5507-
; CHECK-NEXT: lui a3, 1
5508-
; CHECK-NEXT: li a1, 42
5509-
; CHECK-NEXT: slli a4, a2, 32
5510-
; CHECK-NEXT: add a2, a0, a3
5511-
; CHECK-NEXT: srli a3, a4, 32
55125511
; CHECK-NEXT: .LBB119_1: # %vector.body
55135512
; CHECK-NEXT: # =>This Inner Loop Header: Depth=1
5514-
; CHECK-NEXT: vle32.v v9, (a0)
5515-
; CHECK-NEXT: vmseq.vx v0, v9, a1
5516-
; CHECK-NEXT: vsetvli zero, a3, e32, m1, ta, ma
5517-
; CHECK-NEXT: vmerge.vvm v9, v9, v8, v0
5513+
; CHECK-NEXT: vle32.v v8, (a0)
5514+
; CHECK-NEXT: vmseq.vx v0, v8, a3
5515+
; CHECK-NEXT: vsetvli zero, a4, e32, m1, ta, ma
5516+
; CHECK-NEXT: vmerge.vxm v8, v8, a1, v0
55185517
; CHECK-NEXT: vsetivli zero, 4, e32, m1, ta, ma
5519-
; CHECK-NEXT: vse32.v v9, (a0)
5518+
; CHECK-NEXT: vse32.v v8, (a0)
55205519
; CHECK-NEXT: addi a0, a0, 16
55215520
; CHECK-NEXT: bne a0, a2, .LBB119_1
55225521
; CHECK-NEXT: # %bb.2: # %for.cond.cleanup

0 commit comments

Comments
 (0)