llvm/llvm/test/CodeGen/X86/GlobalISel/select-extract-vec512.mir

# RUN: llc -mtriple=x86_64-linux-gnu -mattr=+avx512f -run-pass=instruction-select -verify-machineinstrs %s -o - | FileCheck %s --check-prefix=ALL

# RUN: llc -mtriple=x86_64-linux-gnu -mattr=+avx512f -run-pass=instruction-select -verify-machineinstrs %s -o - | FileCheck %s --check-prefix=ALL

--- |
  define void @test_extract_128_idx0() {
    ret void
  }

  define void @test_extract_128_idx1() {
    ret void
  }

  define void @test_extract_256_idx0() {
    ret void
  }

  define void @test_extract_256_idx1() {
    ret void
  }

...
---
name:            test_extract_128_idx0
# ALL-LABEL: name:  test_extract_128_idx0
alignment:       16
legalized:       true
regBankSelected: true
# ALL:      registers:
# ALL-NEXT:   - { id: 0, class: vr512, preferred-register: '' }
# ALL-NEXT:   - { id: 1, class: vr128x, preferred-register: '' }
registers:
  - { id: 0, class: vecr }
  - { id: 1, class: vecr }
# ALL:          %0:vr512 = COPY $zmm1
# ALL-NEXT:     %1:vr128x = COPY %0.sub_xmm
# ALL-NEXT:     $xmm0 = COPY %1
# ALL-NEXT:     RET 0, implicit $xmm0
body:             |
  bb.1 (%ir-block.0):
    liveins: $zmm1

    %0(<16 x s32>) = COPY $zmm1
    %1(<4 x s32>) = G_EXTRACT %0(<16 x s32>), 0
    $xmm0 = COPY %1(<4 x s32>)
    RET 0, implicit $xmm0

...
---
name:            test_extract_128_idx1
# ALL-LABEL: name:  test_extract_128_idx1
alignment:       16
legalized:       true
regBankSelected: true
# ALL:      registers:
# ALL-NEXT:   - { id: 0, class: vr512, preferred-register: '' }
# ALL-NEXT:   - { id: 1, class: vr128x, preferred-register: '' }
registers:
  - { id: 0, class: vecr }
  - { id: 1, class: vecr }
# ALL:          %0:vr512 = COPY $zmm1
# ALL-NEXT:     %1:vr128x = VEXTRACTF32x4Zrri %0, 1
# ALL-NEXT:     $xmm0 = COPY %1
# ALL-NEXT:     RET 0, implicit $xmm0
body:             |
  bb.1 (%ir-block.0):
    liveins: $zmm1

    %0(<16 x s32>) = COPY $zmm1
    %1(<4 x s32>) = G_EXTRACT %0(<16 x s32>), 128
    $xmm0 = COPY %1(<4 x s32>)
    RET 0, implicit $xmm0

...
---
name:            test_extract_256_idx0
# ALL-LABEL: name:  test_extract_256_idx0
alignment:       16
legalized:       true
regBankSelected: true
# ALL:      registers:
# ALL-NEXT:   - { id: 0, class: vr512, preferred-register: '' }
# ALL-NEXT:   - { id: 1, class: vr256x, preferred-register: '' }
registers:
  - { id: 0, class: vecr }
  - { id: 1, class: vecr }
# ALL:          %0:vr512 = COPY $zmm1
# ALL-NEXT:     %1:vr256x = COPY %0.sub_ymm
# ALL-NEXT:     $ymm0 = COPY %1
# ALL-NEXT:     RET 0, implicit $ymm0
body:             |
  bb.1 (%ir-block.0):
    liveins: $zmm1

    %0(<16 x s32>) = COPY $zmm1
    %1(<8 x s32>) = G_EXTRACT %0(<16 x s32>), 0
    $ymm0 = COPY %1(<8 x s32>)
    RET 0, implicit $ymm0

...
---
name:            test_extract_256_idx1
# ALL-LABEL: name:  test_extract_256_idx1
alignment:       16
legalized:       true
regBankSelected: true
# ALL:      registers:
# ALL-NEXT:   - { id: 0, class: vr512, preferred-register: '' }
# ALL-NEXT:   - { id: 1, class: vr256x, preferred-register: '' }
registers:
  - { id: 0, class: vecr }
  - { id: 1, class: vecr }
# ALL:          %0:vr512 = COPY $zmm1
# ALL-NEXT:     %1:vr256x = VEXTRACTF64x4Zrri %0, 1
# ALL-NEXT:     $ymm0 = COPY %1
# ALL-NEXT:     RET 0, implicit $ymm0
body:             |
  bb.1 (%ir-block.0):
    liveins: $zmm1

    %0(<16 x s32>) = COPY $zmm1
    %1(<8 x s32>) = G_EXTRACT %0(<16 x s32>), 256
    $ymm0 = COPY %1(<8 x s32>)
    RET 0, implicit $ymm0

...