test/CodeGen/AMDGPU/GlobalISel/legalize-extract.mir

# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
# RUN: llc -mtriple=amdgcn-mesa-mesa3d -mcpu=fiji -run-pass=legalizer -global-isel-abort=0 %s -o - | FileCheck %s

---
name: test_extract_lo32_i64
body: |
  bb.0.entry:
    liveins: $vgpr0_vgpr1

    ; CHECK-LABEL: name: test_extract_lo32_i64
    ; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $vgpr0_vgpr1
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s64), 0
    ; CHECK: S_ENDPGM implicit [[EXTRACT]](s32)
    %0:_(s64) = COPY $vgpr0_vgpr1
    %1:_(s32) = G_EXTRACT %0, 0
    S_ENDPGM implicit %1
...
---
name: test_extract_hi32_i64
body: |
  bb.0.entry:
    liveins: $vgpr0_vgpr1

    ; CHECK-LABEL: name: test_extract_hi32_i64
    ; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $vgpr0_vgpr1
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s64), 32
    ; CHECK: S_ENDPGM implicit [[EXTRACT]](s32)
    %0:_(s64) = COPY $vgpr0_vgpr1
    %1:_(s32) = G_EXTRACT %0, 32
     S_ENDPGM implicit %1
...

---
name: test_extract_dword0_i128
body: |
  bb.0.entry:
    liveins: $vgpr0_vgpr1

    ; CHECK-LABEL: name: test_extract_dword0_i128
    ; CHECK: [[COPY:%[0-9]+]]:_(s64) = COPY $vgpr0_vgpr1
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s64), 0
    ; CHECK: S_ENDPGM implicit [[EXTRACT]](s32)
    %0:_(s64) = COPY $vgpr0_vgpr1
    %1:_(s32) = G_EXTRACT %0, 0
    S_ENDPGM implicit %1
...

---
name: test_extract_dword1_i128
body: |
  bb.0.entry:
    liveins: $vgpr0_vgpr1_vgpr2_vgpr3

    ; CHECK-LABEL: name: test_extract_dword1_i128
    ; CHECK: [[COPY:%[0-9]+]]:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s128), 32
    ; CHECK: S_ENDPGM implicit [[EXTRACT]](s32)
    %0:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
    %1:_(s32) = G_EXTRACT %0, 32
    S_ENDPGM implicit %1
...

---
name: test_extract_dword2_i128
body: |
  bb.0.entry:
    liveins: $vgpr0_vgpr1_vgpr2_vgpr3

    ; CHECK-LABEL: name: test_extract_dword2_i128
    ; CHECK: [[COPY:%[0-9]+]]:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s128), 64
    ; CHECK: S_ENDPGM implicit [[EXTRACT]](s32)
    %0:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
    %1:_(s32) = G_EXTRACT %0, 64
    S_ENDPGM implicit %1
...

---
name: test_extract_dword3_i128
body: |
  bb.0.entry:
    liveins: $vgpr0_vgpr1_vgpr2_vgpr3

    ; CHECK-LABEL: name: test_extract_dword3_i128
    ; CHECK: [[COPY:%[0-9]+]]:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s128), 96
    ; CHECK: S_ENDPGM implicit [[EXTRACT]](s32)
    %0:_(s128) = COPY $vgpr0_vgpr1_vgpr2_vgpr3
    %1:_(s32) = G_EXTRACT %0, 96
     S_ENDPGM implicit %1
...
---
name: test_extract_dword2_i96
body: |
  bb.0.entry:
    liveins: $vgpr0_vgpr1_vgpr2

    ; CHECK-LABEL: name: test_extract_dword2_i96
    ; CHECK: [[COPY:%[0-9]+]]:_(s96) = COPY $vgpr0_vgpr1_vgpr2
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s32) = G_EXTRACT [[COPY]](s96), 64
    ; CHECK: S_ENDPGM implicit [[EXTRACT]](s32)
    %0:_(s96) = COPY $vgpr0_vgpr1_vgpr2
    %1:_(s32) = G_EXTRACT %0, 64
    S_ENDPGM implicit %1
...

---
name: extract_s8_v4s8_offset0
legalized: true

body: |
  bb.0:
    ; CHECK-LABEL: name: extract_s8_v4s8_offset0
    ; CHECK: [[DEF:%[0-9]+]]:_(<4 x s32>) = G_IMPLICIT_DEF
    ; CHECK: [[TRUNC:%[0-9]+]]:_(<4 x s8>) = G_TRUNC [[DEF]](<4 x s32>)
    ; CHECK: [[ANYEXT:%[0-9]+]]:_(<4 x s16>) = G_ANYEXT [[TRUNC]](<4 x s8>)
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s16) = G_EXTRACT [[ANYEXT]](<4 x s16>), 0
    ; CHECK: [[TRUNC1:%[0-9]+]]:_(s8) = G_TRUNC [[EXTRACT]](s16)
    ; CHECK: [[ANYEXT1:%[0-9]+]]:_(s32) = G_ANYEXT [[TRUNC1]](s8)
    ; CHECK: $vgpr0 = COPY [[ANYEXT1]](s32)
    %0:_(<4 x s8>) = G_IMPLICIT_DEF
    %1:_(s8) = G_EXTRACT %0, 0
    %2:_(s32) = G_ANYEXT %1
    $vgpr0 = COPY %2
...

---
name: extract_s8_v4s8_offset8
legalized: true

body: |
  bb.0:
    ; CHECK-LABEL: name: extract_s8_v4s8_offset8
    ; CHECK: [[DEF:%[0-9]+]]:_(<4 x s32>) = G_IMPLICIT_DEF
    ; CHECK: [[TRUNC:%[0-9]+]]:_(<4 x s8>) = G_TRUNC [[DEF]](<4 x s32>)
    ; CHECK: [[ANYEXT:%[0-9]+]]:_(<4 x s16>) = G_ANYEXT [[TRUNC]](<4 x s8>)
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s16) = G_EXTRACT [[ANYEXT]](<4 x s16>), 16
    ; CHECK: [[TRUNC1:%[0-9]+]]:_(s8) = G_TRUNC [[EXTRACT]](s16)
    ; CHECK: [[ANYEXT1:%[0-9]+]]:_(s32) = G_ANYEXT [[TRUNC1]](s8)
    ; CHECK: $vgpr0 = COPY [[ANYEXT1]](s32)
    %0:_(<4 x s8>) = G_IMPLICIT_DEF
    %1:_(s8) = G_EXTRACT %0, 8
    %2:_(s32) = G_ANYEXT %1
    $vgpr0 = COPY %2
...

---
name: extract_s8_v4s8_offset16
legalized: true

body: |
  bb.0:
    ; CHECK-LABEL: name: extract_s8_v4s8_offset16
    ; CHECK: [[DEF:%[0-9]+]]:_(<4 x s32>) = G_IMPLICIT_DEF
    ; CHECK: [[TRUNC:%[0-9]+]]:_(<4 x s8>) = G_TRUNC [[DEF]](<4 x s32>)
    ; CHECK: [[ANYEXT:%[0-9]+]]:_(<4 x s16>) = G_ANYEXT [[TRUNC]](<4 x s8>)
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s16) = G_EXTRACT [[ANYEXT]](<4 x s16>), 32
    ; CHECK: [[TRUNC1:%[0-9]+]]:_(s8) = G_TRUNC [[EXTRACT]](s16)
    ; CHECK: [[ANYEXT1:%[0-9]+]]:_(s32) = G_ANYEXT [[TRUNC1]](s8)
    ; CHECK: $vgpr0 = COPY [[ANYEXT1]](s32)
    %0:_(<4 x s8>) = G_IMPLICIT_DEF
    %1:_(s8) = G_EXTRACT %0, 16
    %2:_(s32) = G_ANYEXT %1
    $vgpr0 = COPY %2
...

---
name: extract_s8_v4s8_offset24
legalized: true

body: |
  bb.0:
    ; CHECK-LABEL: name: extract_s8_v4s8_offset24
    ; CHECK: [[DEF:%[0-9]+]]:_(<4 x s32>) = G_IMPLICIT_DEF
    ; CHECK: [[TRUNC:%[0-9]+]]:_(<4 x s8>) = G_TRUNC [[DEF]](<4 x s32>)
    ; CHECK: [[ANYEXT:%[0-9]+]]:_(<4 x s16>) = G_ANYEXT [[TRUNC]](<4 x s8>)
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(s16) = G_EXTRACT [[ANYEXT]](<4 x s16>), 48
    ; CHECK: [[TRUNC1:%[0-9]+]]:_(s8) = G_TRUNC [[EXTRACT]](s16)
    ; CHECK: [[ANYEXT1:%[0-9]+]]:_(s32) = G_ANYEXT [[TRUNC1]](s8)
    ; CHECK: $vgpr0 = COPY [[ANYEXT1]](s32)
    %0:_(<4 x s8>) = G_IMPLICIT_DEF
    %1:_(s8) = G_EXTRACT %0, 24
    %2:_(s32) = G_ANYEXT %1
    $vgpr0 = COPY %2
...

---
name: extract_s8_v3s8_offset16
legalized: true

body: |
  bb.0:
    ; CHECK-LABEL: name: extract_s8_v3s8_offset16
    ; CHECK: [[DEF:%[0-9]+]]:_(<4 x s32>) = G_IMPLICIT_DEF
    ; CHECK: [[TRUNC:%[0-9]+]]:_(<4 x s8>) = G_TRUNC [[DEF]](<4 x s32>)
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(<3 x s8>) = G_EXTRACT [[TRUNC]](<4 x s8>), 0
    ; CHECK: [[ANYEXT:%[0-9]+]]:_(<3 x s16>) = G_ANYEXT [[EXTRACT]](<3 x s8>)
    ; CHECK: [[EXTRACT1:%[0-9]+]]:_(s16) = G_EXTRACT [[ANYEXT]](<3 x s16>), 32
    ; CHECK: [[TRUNC1:%[0-9]+]]:_(s8) = G_TRUNC [[EXTRACT1]](s16)
    ; CHECK: [[ANYEXT1:%[0-9]+]]:_(s32) = G_ANYEXT [[TRUNC1]](s8)
    ; CHECK: $vgpr0 = COPY [[ANYEXT1]](s32)
    %0:_(<3 x s8>) = G_IMPLICIT_DEF
    %1:_(s8) = G_EXTRACT %0, 16
    %2:_(s32) = G_ANYEXT %1
    $vgpr0 = COPY %2
...

---
name: extract_s8_v5s1_offset4
legalized: true

body: |
  bb.0:
    ; CHECK-LABEL: name: extract_s8_v5s1_offset4
    ; CHECK: [[DEF:%[0-9]+]]:_(<6 x s32>) = G_IMPLICIT_DEF
    ; CHECK: [[TRUNC:%[0-9]+]]:_(<6 x s1>) = G_TRUNC [[DEF]](<6 x s32>)
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(<5 x s1>) = G_EXTRACT [[TRUNC]](<6 x s1>), 0
    ; CHECK: [[ANYEXT:%[0-9]+]]:_(<5 x s16>) = G_ANYEXT [[EXTRACT]](<5 x s1>)
    ; CHECK: [[EXTRACT1:%[0-9]+]]:_(s16) = G_EXTRACT [[ANYEXT]](<5 x s16>), 64
    ; CHECK: [[TRUNC1:%[0-9]+]]:_(s1) = G_TRUNC [[EXTRACT1]](s16)
    ; CHECK: [[ANYEXT1:%[0-9]+]]:_(s32) = G_ANYEXT [[TRUNC1]](s1)
    ; CHECK: $vgpr0 = COPY [[ANYEXT1]](s32)
    %0:_(<5 x s1>) = G_IMPLICIT_DEF
    %1:_(s1) = G_EXTRACT %0, 4
    %2:_(s32) = G_ANYEXT %1
    $vgpr0 = COPY %2
...

---
name: extract_v2s16_v4s16_offset32
legalized: true

body: |
  bb.0:
    ; CHECK-LABEL: name: extract_v2s16_v4s16_offset32
    ; CHECK: [[DEF:%[0-9]+]]:_(<4 x s16>) = G_IMPLICIT_DEF
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(<2 x s16>) = G_EXTRACT [[DEF]](<4 x s16>), 32
    ; CHECK: $vgpr0 = COPY [[EXTRACT]](<2 x s16>)
    %0:_(<4 x s16>) = G_IMPLICIT_DEF
    %1:_(<2 x s16>) = G_EXTRACT %0, 32
    $vgpr0 = COPY %1
...

---
name: extract_v2s16_v6s16_offset32
legalized: true

body: |
  bb.0:
    ; CHECK-LABEL: name: extract_v2s16_v6s16_offset32
    ; CHECK: [[DEF:%[0-9]+]]:_(<6 x s32>) = G_IMPLICIT_DEF
    ; CHECK: [[TRUNC:%[0-9]+]]:_(<6 x s16>) = G_TRUNC [[DEF]](<6 x s32>)
    ; CHECK: [[EXTRACT:%[0-9]+]]:_(<2 x s16>) = G_EXTRACT [[TRUNC]](<6 x s16>), 32
    ; CHECK: $vgpr0 = COPY [[EXTRACT]](<2 x s16>)
    %0:_(<6 x s16>) = G_IMPLICIT_DEF
    %1:_(<2 x s16>) = G_EXTRACT %0, 32
    $vgpr0 = COPY %1
...