# NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
# RUN: llc -o - -mtriple=aarch64-unknown-unknown -run-pass=aarch64-prelegalizer-combiner -verify-machineinstrs %s | FileCheck %s --check-prefixes=CHECK,CHECK-PRE
# RUN: llc -o - -mtriple=aarch64-unknown-unknown -run-pass=aarch64-postlegalizer-combiner -verify-machineinstrs %s | FileCheck %s --check-prefixes=CHECK,CHECK-POST
---
name: test_combine_trunc_select
legalized: true
body: |
bb.1:
; CHECK-PRE-LABEL: name: test_combine_trunc_select
; CHECK-PRE: %cond:_(s32) = COPY $w0
; CHECK-PRE-NEXT: %lhs:_(s64) = COPY $x0
; CHECK-PRE-NEXT: %rhs:_(s64) = COPY $x0
; CHECK-PRE-NEXT: [[TRUNC:%[0-9]+]]:_(s32) = G_TRUNC %lhs(s64)
; CHECK-PRE-NEXT: [[TRUNC1:%[0-9]+]]:_(s32) = G_TRUNC %rhs(s64)
; CHECK-PRE-NEXT: %small:_(s32) = G_SELECT %cond(s32), [[TRUNC]], [[TRUNC1]]
; CHECK-PRE-NEXT: $w0 = COPY %small(s32)
;
; CHECK-POST-LABEL: name: test_combine_trunc_select
; CHECK-POST: %cond:_(s32) = COPY $w0
; CHECK-POST-NEXT: %lhs:_(s64) = COPY $x0
; CHECK-POST-NEXT: %rhs:_(s64) = COPY $x0
; CHECK-POST-NEXT: %res:_(s64) = G_SELECT %cond(s32), %lhs, %rhs
; CHECK-POST-NEXT: %small:_(s32) = G_TRUNC %res(s64)
; CHECK-POST-NEXT: $w0 = COPY %small(s32)
%cond:_(s32) = COPY $w0
%lhs:_(s64) = COPY $x0
%rhs:_(s64) = COPY $x0
%res:_(s64) = G_SELECT %cond(s32), %lhs, %rhs
%small:_(s32) = G_TRUNC %res(s64)
$w0 = COPY %small(s32)
...
---
name: test_combine_zext_select
legalized: true
body: |
bb.1:
; CHECK-PRE-LABEL: name: test_combine_zext_select
; CHECK-PRE: %cond:_(s32) = COPY $w0
; CHECK-PRE-NEXT: %lhs:_(s32) = COPY $w0
; CHECK-PRE-NEXT: %rhs:_(s32) = COPY $w0
; CHECK-PRE-NEXT: [[ZEXT:%[0-9]+]]:_(s64) = G_ZEXT %lhs(s32)
; CHECK-PRE-NEXT: [[ZEXT1:%[0-9]+]]:_(s64) = G_ZEXT %rhs(s32)
; CHECK-PRE-NEXT: %big:_(s64) = G_SELECT %cond(s32), [[ZEXT]], [[ZEXT1]]
; CHECK-PRE-NEXT: $x0 = COPY %big(s64)
;
; CHECK-POST-LABEL: name: test_combine_zext_select
; CHECK-POST: %cond:_(s32) = COPY $w0
; CHECK-POST-NEXT: %lhs:_(s32) = COPY $w0
; CHECK-POST-NEXT: %rhs:_(s32) = COPY $w0
; CHECK-POST-NEXT: %res:_(s32) = G_SELECT %cond(s32), %lhs, %rhs
; CHECK-POST-NEXT: %big:_(s64) = G_ZEXT %res(s32)
; CHECK-POST-NEXT: $x0 = COPY %big(s64)
%cond:_(s32) = COPY $w0
%lhs:_(s32) = COPY $w0
%rhs:_(s32) = COPY $w0
%res:_(s32) = G_SELECT %cond(s32), %lhs, %rhs
%big:_(s64) = G_ZEXT %res(s32)
$x0 = COPY %big(s64)
...
---
name: test_combine_anyzext_select
legalized: true
body: |
bb.1:
; CHECK-PRE-LABEL: name: test_combine_anyzext_select
; CHECK-PRE: %cond:_(s32) = COPY $w0
; CHECK-PRE-NEXT: %lhs:_(s32) = COPY $w0
; CHECK-PRE-NEXT: %rhs:_(s32) = COPY $w0
; CHECK-PRE-NEXT: [[ANYEXT:%[0-9]+]]:_(s64) = G_ANYEXT %lhs(s32)
; CHECK-PRE-NEXT: [[ANYEXT1:%[0-9]+]]:_(s64) = G_ANYEXT %rhs(s32)
; CHECK-PRE-NEXT: %big:_(s64) = G_SELECT %cond(s32), [[ANYEXT]], [[ANYEXT1]]
; CHECK-PRE-NEXT: $x0 = COPY %big(s64)
;
; CHECK-POST-LABEL: name: test_combine_anyzext_select
; CHECK-POST: %cond:_(s32) = COPY $w0
; CHECK-POST-NEXT: %lhs:_(s32) = COPY $w0
; CHECK-POST-NEXT: %rhs:_(s32) = COPY $w0
; CHECK-POST-NEXT: %res:_(s32) = G_SELECT %cond(s32), %lhs, %rhs
; CHECK-POST-NEXT: %big:_(s64) = G_ANYEXT %res(s32)
; CHECK-POST-NEXT: $x0 = COPY %big(s64)
%cond:_(s32) = COPY $w0
%lhs:_(s32) = COPY $w0
%rhs:_(s32) = COPY $w0
%res:_(s32) = G_SELECT %cond(s32), %lhs, %rhs
%big:_(s64) = G_ANYEXT %res(s32)
$x0 = COPY %big(s64)
...
---
name: test_combine_anyzext_select_multi_use
legalized: true
body: |
bb.1:
; CHECK-LABEL: name: test_combine_anyzext_select_multi_use
; CHECK: %cond:_(s32) = COPY $w0
; CHECK-NEXT: %lhs:_(s32) = COPY $w0
; CHECK-NEXT: %rhs:_(s32) = COPY $w0
; CHECK-NEXT: %res:_(s32) = G_SELECT %cond(s32), %lhs, %rhs
; CHECK-NEXT: %big:_(s64) = G_ANYEXT %res(s32)
; CHECK-NEXT: $x0 = COPY %big(s64)
; CHECK-NEXT: $w0 = COPY %res(s32)
%cond:_(s32) = COPY $w0
%lhs:_(s32) = COPY $w0
%rhs:_(s32) = COPY $w0
%res:_(s32) = G_SELECT %cond(s32), %lhs, %rhs
%big:_(s64) = G_ANYEXT %res(s32)
$x0 = COPY %big(s64)
$w0 = COPY %res(s32)
...
---
name: test_combine_trunc_select_vector_out_of_budget
legalized: true
body: |
bb.1:
; CHECK-LABEL: name: test_combine_trunc_select_vector_out_of_budget
; CHECK: %cond:_(<2 x s32>) = COPY $x0
; CHECK-NEXT: %arg1:_(s64) = COPY $x0
; CHECK-NEXT: %arg2:_(s64) = COPY $x0
; CHECK-NEXT: %bv:_(<2 x s64>) = G_BUILD_VECTOR %arg1(s64), %arg2(s64)
; CHECK-NEXT: %bv2:_(<2 x s64>) = G_BUILD_VECTOR %arg2(s64), %arg1(s64)
; CHECK-NEXT: %res:_(<2 x s64>) = G_SELECT %cond(<2 x s32>), %bv, %bv2
; CHECK-NEXT: %small:_(<2 x s32>) = G_TRUNC %res(<2 x s64>)
; CHECK-NEXT: $x0 = COPY %small(<2 x s32>)
%cond:_(<2 x s32>) = COPY $x0
%arg1:_(s64) = COPY $x0
%arg2:_(s64) = COPY $x0
%bv:_(<2 x s64>) = G_BUILD_VECTOR %arg1(s64), %arg2(s64)
%bv2:_(<2 x s64>) = G_BUILD_VECTOR %arg2(s64), %arg1(s64)
%res:_(<2 x s64>) = G_SELECT %cond(<2 x s32>), %bv, %bv2
%small:_(<2 x s32>) = G_TRUNC %res(<2 x s64>)
$x0 = COPY %small(<2 x s32>)
...
---
name: test_combine_trunc_build_vector
legalized: true
body: |
bb.1:
; CHECK-PRE-LABEL: name: test_combine_trunc_build_vector
; CHECK-PRE: %arg1:_(s64) = COPY $x0
; CHECK-PRE-NEXT: %arg2:_(s64) = COPY $x0
; CHECK-PRE-NEXT: [[TRUNC:%[0-9]+]]:_(s32) = G_TRUNC %arg1(s64)
; CHECK-PRE-NEXT: [[TRUNC1:%[0-9]+]]:_(s32) = G_TRUNC %arg2(s64)
; CHECK-PRE-NEXT: %small:_(<2 x s32>) = G_BUILD_VECTOR [[TRUNC]](s32), [[TRUNC1]](s32)
; CHECK-PRE-NEXT: $x0 = COPY %small(<2 x s32>)
;
; CHECK-POST-LABEL: name: test_combine_trunc_build_vector
; CHECK-POST: %arg1:_(s64) = COPY $x0
; CHECK-POST-NEXT: %arg2:_(s64) = COPY $x0
; CHECK-POST-NEXT: %bv:_(<2 x s64>) = G_BUILD_VECTOR %arg1(s64), %arg2(s64)
; CHECK-POST-NEXT: %small:_(<2 x s32>) = G_TRUNC %bv(<2 x s64>)
; CHECK-POST-NEXT: $x0 = COPY %small(<2 x s32>)
%arg1:_(s64) = COPY $x0
%arg2:_(s64) = COPY $x0
%bv:_(<2 x s64>) = G_BUILD_VECTOR %arg1(s64), %arg2(s64)
%small:_(<2 x s32>) = G_TRUNC %bv(<2 x s64>)
$x0 = COPY %small(<2 x s32>)
...
---
name: test_combine_zext_build_vector
legalized: true
body: |
bb.1:
; CHECK-LABEL: name: test_combine_zext_build_vector
; CHECK: %arg1:_(s32) = COPY $w0
; CHECK-NEXT: %arg2:_(s32) = COPY $w0
; CHECK-NEXT: %bv:_(<2 x s32>) = G_BUILD_VECTOR %arg1(s32), %arg2(s32)
; CHECK-NEXT: %large:_(<2 x s64>) = G_ZEXT %bv(<2 x s32>)
; CHECK-NEXT: $q0 = COPY %large(<2 x s64>)
%arg1:_(s32) = COPY $w0
%arg2:_(s32) = COPY $w0
%bv:_(<2 x s32>) = G_BUILD_VECTOR %arg1(s32), %arg2(s32)
%large:_(<2 x s64>) = G_ZEXT %bv(<2 x s32>)
$q0 = COPY %large(<2 x s64>)
...
---
name: test_combine_anyext_build_vector
legalized: true
body: |
bb.1:
; CHECK-LABEL: name: test_combine_anyext_build_vector
; CHECK: %arg1:_(s32) = COPY $w0
; CHECK-NEXT: %arg2:_(s32) = COPY $w0
; CHECK-NEXT: %bv:_(<2 x s32>) = G_BUILD_VECTOR %arg1(s32), %arg2(s32)
; CHECK-NEXT: %large:_(<2 x s64>) = G_ANYEXT %bv(<2 x s32>)
; CHECK-NEXT: $q0 = COPY %large(<2 x s64>)
%arg1:_(s32) = COPY $w0
%arg2:_(s32) = COPY $w0
%bv:_(<2 x s32>) = G_BUILD_VECTOR %arg1(s32), %arg2(s32)
%large:_(<2 x s64>) = G_ANYEXT %bv(<2 x s32>)
$q0 = COPY %large(<2 x s64>)
...
---
name: test_combine_sext_build_vector
legalized: true
body: |
bb.1:
; CHECK-LABEL: name: test_combine_sext_build_vector
; CHECK: %arg1:_(s32) = COPY $w0
; CHECK-NEXT: %arg2:_(s32) = COPY $w0
; CHECK-NEXT: %bv:_(<2 x s32>) = G_BUILD_VECTOR %arg1(s32), %arg2(s32)
; CHECK-NEXT: %large:_(<2 x s64>) = G_SEXT %bv(<2 x s32>)
; CHECK-NEXT: $q0 = COPY %large(<2 x s64>)
%arg1:_(s32) = COPY $w0
%arg2:_(s32) = COPY $w0
%bv:_(<2 x s32>) = G_BUILD_VECTOR %arg1(s32), %arg2(s32)
%large:_(<2 x s64>) = G_SEXT %bv(<2 x s32>)
$q0 = COPY %large(<2 x s64>)
...
---
name: test_combine_anyext_build_vector_multi_use
legalized: true
body: |
bb.1:
; CHECK-LABEL: name: test_combine_anyext_build_vector_multi_use
; CHECK: %arg1:_(s32) = COPY $w0
; CHECK-NEXT: %arg2:_(s32) = COPY $w0
; CHECK-NEXT: %bv:_(<2 x s32>) = G_BUILD_VECTOR %arg1(s32), %arg2(s32)
; CHECK-NEXT: %large:_(<2 x s64>) = G_ANYEXT %bv(<2 x s32>)
; CHECK-NEXT: $q0 = COPY %large(<2 x s64>)
; CHECK-NEXT: $d0 = COPY %bv(<2 x s32>)
%arg1:_(s32) = COPY $w0
%arg2:_(s32) = COPY $w0
%bv:_(<2 x s32>) = G_BUILD_VECTOR %arg1(s32), %arg2(s32)
%large:_(<2 x s64>) = G_ANYEXT %bv(<2 x s32>)
$q0 = COPY %large(<2 x s64>)
$d0 = COPY %bv(<2 x s32>)