src/simd/archsimd/_gen/simdgen/ops/ShiftRotate/go_arm64.yaml

!sum

# Shift by vector
- go: Shift
  signed: true
  asm: "VSSHL"
  in:
  - &int
    go: $t
    base: int
  - go: $t
    name: shift
  out:
  - *int

- go: Shift
  signed: false
  asm: "VUSHL"
  in:
  - &uint
    go: $t
    base: uint
  - go: $t
    base: uint
    overwriteBase: int
    name: shift
  out:
  - *uint

- go: ScaleSaturated
  signed: true
  asm: "VSQSHL"
  in:
  - *int
  - go: $t
    base: int
    name: scale
  out:
  - *int

- go: ScaleSaturated
  signed: false
  asm: "VUQSHL"
  in:
  - *uint
  - go: $t
    base: uint
    overwriteBase: int
    name: scale
  out:
  - *uint

# Shift right by constant
- go: shiftAllRightConst
  signed: true
  asm: "VSSHR"
  specialLower: "if a==0 => x"
  in:
  - &imm
    class: immediate
    immOffset: 0
  - *int
  out:
  - *int

- go: shiftAllRightConst
  signed: false
  asm: "VUSHR"
  specialLower: "if a==0 => x"
  in:
  - *imm
  - *uint
  out:
  - *uint

# Shift left by constant
- go: shiftAllLeftConst
  signed: true
  asm: "VSHL"
  specialLower: "if a==0 => x"
  in:
  - *imm
  - *int
  out:
  - *int

- go: shiftAllLeftConst
  signed: false
  asm: "VSHL"
  specialLower: "if a==0 => x"
  in:
  - *imm
  - *uint
  out:
  - *uint

# Shift left saturated by constant
- go: shiftLeftSaturatedConst
  signed: true
  asm: "VSQSHL"
  ssaVariant: "const"
  specialLower: "if a==0 => x"
  in:
  - *imm
  - *int
  out:
  - *int

- go: shiftLeftSaturatedConst
  signed: false
  asm: "VUQSHL"
  ssaVariant: "const"
  specialLower: "if a==0 => x"
  in:
  - *imm
  - *uint
  out:
  - *uint

# Shift all lanes in vector x by the same unsigned scalar amount y.
# ShiftAll* are lowered to corresponding ShiftConst* if the amount is constant and fits into immediate,
# otherwise fallback to the Shift intrinsic using sign to specify shift direction, e.g.:
#   x.Shift(Zero.SetElem(0, -y).Broadcast1ToN()) for shift right
#   x.Shift(Zero.SetElem(0,  y).Broadcast1ToN()) for shift left
- go: ShiftAllLeft
  signed: true
  asm: "VSSHL"
  specialLower: !string |-
    earlymatch (%g x y) => (VSSHL%a x (VDUP%sbcast [0] (VMOV%sins [0] x (CSEL <typ.UInt64> [ssaop.OpARM64LessThanU] y (MOVDconst [127]) (CMPconst [127] y)))))

  in:
  - *int
  - &scalarShiftSigned
    go: $t
    base: int
    treatLikeAScalarOfSize: 64
    name: shift
  out:
  - *int

- go: ShiftAllLeft
  signed: false
  asm: "VUSHL"
  specialLower: !string "earlymatch (%g x y) => (VUSHL%a x (VDUP%sbcast [0] (VMOV%sins [0] x (CSEL <typ.UInt64> [ssaop.OpARM64LessThanU] y (MOVDconst [127]) (CMPconst [127] y)))))"
  in:
  - *uint
  - &scalarShiftUnsigned
    go: $t
    base: uint
    treatLikeAScalarOfSize: 64
    overwriteBase: int
    name: shift
  out:
  - *uint

- go: ShiftAllRight
  signed: true
  asm: "VSSHL"
  specialLower: !string "earlymatch (%g x y) => (VSSHL%a x (VDUP%sbcast [0] (VMOV%sins [0] x (NEG <typ.Int64> (CSEL <typ.UInt64> [ssaop.OpARM64LessThanU] y (MOVDconst [127]) (CMPconst [127] y))))))"
  in:
  - *int
  - *scalarShiftSigned
  out:
  - *int

- go: ShiftAllRight
  signed: false
  asm: "VUSHL"
  specialLower: !string "earlymatch (%g x y) => (VUSHL%a x (VDUP%sbcast [0] (VMOV%sins [0] x (NEG <typ.Int64> (CSEL <typ.UInt64> [ssaop.OpARM64LessThanU] y (MOVDconst [127]) (CMPconst [127] y))))))"
  in:
  - *uint
  - *scalarShiftUnsigned
  out:
  - *uint

- go: shiftRightNarrowConst
  asm: "VSHRN"
  hiHalfAsm: "VSHRN2"
  specialLower: !string "match (%h [0] x) => (VXTN%a x)"
  in:
  - *imm
  - go: $t
    base: uint
  out:
  - go: $u
    base: uint

- go: shiftLeftWidenLoConst
  signed: false
  asm: "VUSHLL"
  hiHalfAsm: "VUSHLL2"
  in:
  - *imm
  - go: $t
    base: uint
  out:
  - go: $u
    base: uint

- go: shiftLeftWidenLoConst
  signed: true
  asm: "VSSHLL"
  hiHalfAsm: "VSSHLL2"
  in:
  - *imm
  - go: $t
    base: int
  out:
  - go: $u
    base: int