Files
TheRock/compiler/amd-llvm/llvm/test/CodeGen/AMDGPU/GlobalISel/combine-fma-add-ext-mul.ll
T
derek.holloway 7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

154 lines
7.8 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -global-isel -new-reg-bank-select -mtriple=amdgcn -mcpu=gfx900 --denormal-fp-math=preserve-sign < %s | FileCheck -check-prefix=GFX9-FAST-DENORM %s
; RUN: llc -global-isel -new-reg-bank-select -mtriple=amdgcn -mcpu=gfx1010 --denormal-fp-math=preserve-sign < %s | FileCheck -check-prefix=GFX10-FAST-DENORM %s
; fold (fadd fast (fpext (fmul fast x, y)), z) -> (fma (fpext x), (fpext y), z)
; fold (fadd fast x, (fpext (fmul fast y, z))) -> (fma (fpext y), (fpext z), x)
define amdgpu_vs float @test_f16_f32_add_ext_mul(half inreg %x, half inreg %y, float inreg %z) {
; GFX9-FAST-DENORM-LABEL: test_f16_f32_add_ext_mul:
; GFX9-FAST-DENORM: ; %bb.0: ; %.entry
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s1
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v1, s2
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v0, s0, v0, v1 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: ; return to shader part epilog
;
; GFX10-FAST-DENORM-LABEL: test_f16_f32_add_ext_mul:
; GFX10-FAST-DENORM: ; %bb.0: ; %.entry
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s2
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v0, s0, s1, v0 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: ; return to shader part epilog
.entry:
%a = fmul fast half %x, %y
%b = fpext half %a to float
%c = fadd fast float %b, %z
ret float %c
}
define amdgpu_vs float @test_f16_f32_add_ext_mul_rhs(half inreg %x, half inreg %y, float inreg %z) {
; GFX9-FAST-DENORM-LABEL: test_f16_f32_add_ext_mul_rhs:
; GFX9-FAST-DENORM: ; %bb.0: ; %.entry
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s1
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v1, s2
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v0, s0, v0, v1 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: ; return to shader part epilog
;
; GFX10-FAST-DENORM-LABEL: test_f16_f32_add_ext_mul_rhs:
; GFX10-FAST-DENORM: ; %bb.0: ; %.entry
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s2
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v0, s0, s1, v0 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: ; return to shader part epilog
.entry:
%a = fmul fast half %x, %y
%b = fpext half %a to float
%c = fadd fast float %z, %b
ret float %c
}
define amdgpu_vs <5 x float> @test_5xf16_5xf32_add_ext_mul(<5 x half> inreg %x, <5 x half> inreg %y, <5 x float> inreg %z) {
; GFX9-FAST-DENORM-LABEL: test_5xf16_5xf32_add_ext_mul:
; GFX9-FAST-DENORM: ; %bb.0: ; %.entry
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s3
; GFX9-FAST-DENORM-NEXT: v_pk_mul_f16 v0, s0, v0
; GFX9-FAST-DENORM-NEXT: v_readfirstlane_b32 s0, v0
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s4
; GFX9-FAST-DENORM-NEXT: v_pk_mul_f16 v0, s1, v0
; GFX9-FAST-DENORM-NEXT: v_readfirstlane_b32 s1, v0
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s5
; GFX9-FAST-DENORM-NEXT: v_pk_mul_f16 v0, s2, v0
; GFX9-FAST-DENORM-NEXT: v_readfirstlane_b32 s2, v0
; GFX9-FAST-DENORM-NEXT: s_lshr_b32 s3, s0, 16
; GFX9-FAST-DENORM-NEXT: s_lshr_b32 s4, s1, 16
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s6
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v1, s7
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v2, s8
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v3, s9
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v4, s10
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v0, s0, 1.0, v0 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v1, s3, 1.0, v1 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v2, s1, 1.0, v2 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v3, s4, 1.0, v3 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v4, s2, 1.0, v4 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: ; return to shader part epilog
;
; GFX10-FAST-DENORM-LABEL: test_5xf16_5xf32_add_ext_mul:
; GFX10-FAST-DENORM: ; %bb.0: ; %.entry
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s6
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v1, s7
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v2, s8
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v3, s9
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v4, s10
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s11, s0, 16
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s12, s1, 16
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s13, s3, 16
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s14, s4, 16
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v0, s0, s3, v0 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v1, s11, s13, v1 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v2, s1, s4, v2 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v3, s12, s14, v3 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v4, s2, s5, v4 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: ; return to shader part epilog
.entry:
%a = fmul fast <5 x half> %x, %y
%b = fpext <5 x half> %a to <5 x float>
%c = fadd fast <5 x float> %b, %z
ret <5 x float> %c
}
define amdgpu_vs <6 x float> @test_6xf16_6xf32_add_ext_mul_rhs(<6 x half> inreg %x, <6 x half> inreg %y, <6 x float> inreg %z) {
; GFX9-FAST-DENORM-LABEL: test_6xf16_6xf32_add_ext_mul_rhs:
; GFX9-FAST-DENORM: ; %bb.0: ; %.entry
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s3
; GFX9-FAST-DENORM-NEXT: v_pk_mul_f16 v0, s0, v0
; GFX9-FAST-DENORM-NEXT: v_readfirstlane_b32 s0, v0
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s4
; GFX9-FAST-DENORM-NEXT: v_pk_mul_f16 v0, s1, v0
; GFX9-FAST-DENORM-NEXT: v_readfirstlane_b32 s1, v0
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s5
; GFX9-FAST-DENORM-NEXT: v_pk_mul_f16 v0, s2, v0
; GFX9-FAST-DENORM-NEXT: v_readfirstlane_b32 s2, v0
; GFX9-FAST-DENORM-NEXT: s_lshr_b32 s3, s0, 16
; GFX9-FAST-DENORM-NEXT: s_lshr_b32 s4, s1, 16
; GFX9-FAST-DENORM-NEXT: s_lshr_b32 s5, s2, 16
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s6
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v1, s7
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v2, s8
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v3, s9
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v4, s10
; GFX9-FAST-DENORM-NEXT: v_mov_b32_e32 v5, s11
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v0, s0, 1.0, v0 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v1, s3, 1.0, v1 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v2, s1, 1.0, v2 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v3, s4, 1.0, v3 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v4, s2, 1.0, v4 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: v_mad_mix_f32 v5, s5, 1.0, v5 op_sel_hi:[1,1,0]
; GFX9-FAST-DENORM-NEXT: ; return to shader part epilog
;
; GFX10-FAST-DENORM-LABEL: test_6xf16_6xf32_add_ext_mul_rhs:
; GFX10-FAST-DENORM: ; %bb.0: ; %.entry
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v0, s6
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v1, s7
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v2, s8
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v3, s9
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v4, s10
; GFX10-FAST-DENORM-NEXT: v_mov_b32_e32 v5, s11
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s12, s0, 16
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s13, s1, 16
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s6, s2, 16
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s14, s3, 16
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s15, s4, 16
; GFX10-FAST-DENORM-NEXT: s_lshr_b32 s16, s5, 16
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v0, s0, s3, v0 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v1, s12, s14, v1 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v2, s1, s4, v2 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v3, s13, s15, v3 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v4, s2, s5, v4 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: v_fma_mix_f32 v5, s6, s16, v5 op_sel_hi:[1,1,0]
; GFX10-FAST-DENORM-NEXT: ; return to shader part epilog
.entry:
%a = fmul fast <6 x half> %x, %y
%b = fpext <6 x half> %a to <6 x float>
%c = fadd fast <6 x float> %z, %b
ret <6 x float> %c
}