https://github.com/arsenm updated https://github.com/llvm/llvm-project/pull/142114
>From da7b0574d489d67f6f05dd396e4a8bdf95941bf8 Mon Sep 17 00:00:00 2001 From: Matt Arsenault <matthew.arsena...@amd.com> Date: Fri, 30 May 2025 11:21:42 +0200 Subject: [PATCH 1/2] AMDGPU: Move bf16 copysign tests to separate file Make symmetric with other copysign tests --- llvm/test/CodeGen/AMDGPU/fcopysign.bf16.ll | 959 +++++++++++++++++++++ 1 file changed, 959 insertions(+) create mode 100644 llvm/test/CodeGen/AMDGPU/fcopysign.bf16.ll diff --git a/llvm/test/CodeGen/AMDGPU/fcopysign.bf16.ll b/llvm/test/CodeGen/AMDGPU/fcopysign.bf16.ll new file mode 100644 index 0000000000000..4fcce8a6d623f --- /dev/null +++ b/llvm/test/CodeGen/AMDGPU/fcopysign.bf16.ll @@ -0,0 +1,959 @@ +; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5 +; RUN: llc < %s -mtriple=amdgcn | FileCheck %s -check-prefixes=GCN +; RUN: llc < %s -mtriple=amdgcn -mcpu=hawaii | FileCheck %s -check-prefixes=GFX7 +; RUN: llc < %s -mtriple=amdgcn -mcpu=tonga | FileCheck %s -check-prefixes=GFX8 +; RUN: llc < %s -mtriple=amdgcn -mcpu=gfx900 | FileCheck %s -check-prefixes=GFX9 +; RUN: llc < %s -mtriple=amdgcn -mcpu=gfx1010 | FileCheck %s -check-prefixes=GFX10 +; RUN: llc < %s -mtriple=amdgcn -mcpu=gfx1100 -mattr=+real-true16 | FileCheck %s -check-prefixes=GFX11,GFX11TRUE16 +; RUN: llc < %s -mtriple=amdgcn -mcpu=gfx1100 -mattr=-real-true16 | FileCheck %s -check-prefixes=GFX11,GFX11FAKE16 + +declare bfloat @llvm.copysign.bf16(bfloat, bfloat) + +define bfloat @v_copysign_bf16_bf16(bfloat %mag, bfloat %sign) { +; GCN-LABEL: v_copysign_bf16_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GCN-NEXT: v_and_b32_e32 v1, 0x80000000, v1 +; GCN-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, v0, v1 +; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_bf16_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GFX7-NEXT: v_and_b32_e32 v1, 0x80000000, v1 +; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GFX7-NEXT: v_or_b32_e32 v0, v0, v1 +; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_bf16_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: s_movk_i32 s4, 0x7fff +; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_bf16_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: s_movk_i32 s4, 0x7fff +; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_bf16_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_bf16_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + ret bfloat %op +} + +define bfloat @v_copysign_bf16_s_bf16(bfloat %mag, bfloat inreg %sign) { +; GCN-LABEL: v_copysign_bf16_s_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GCN-NEXT: s_and_b32 s4, s16, 0x80000000 +; GCN-NEXT: s_lshr_b32 s4, s4, 16 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, s4, v0 +; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_bf16_s_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GFX7-NEXT: s_and_b32 s4, s16, 0x80000000 +; GFX7-NEXT: s_lshr_b32 s4, s4, 16 +; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GFX7-NEXT: v_or_b32_e32 v0, s4, v0 +; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_bf16_s_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: s_movk_i32 s4, 0x7fff +; GFX8-NEXT: v_mov_b32_e32 v1, s16 +; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_bf16_s_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: s_movk_i32 s4, 0x7fff +; GFX9-NEXT: v_mov_b32_e32 v1, s16 +; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_bf16_s_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, s16 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_bf16_s_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, s0 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + ret bfloat %op +} + +define bfloat @v_copysign_s_bf16_bf16(bfloat inreg %mag, bfloat %sign) { +; GCN-LABEL: v_copysign_s_bf16_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: v_mul_f32_e64 v1, 1.0, s16 +; GCN-NEXT: v_and_b32_e32 v0, 0x80000000, v0 +; GCN-NEXT: v_lshrrev_b32_e32 v0, 16, v0 +; GCN-NEXT: v_bfe_u32 v1, v1, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, v1, v0 +; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_s_bf16_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: v_mul_f32_e64 v1, 1.0, s16 +; GFX7-NEXT: v_and_b32_e32 v0, 0x80000000, v0 +; GFX7-NEXT: v_lshrrev_b32_e32 v0, 16, v0 +; GFX7-NEXT: v_bfe_u32 v1, v1, 16, 15 +; GFX7-NEXT: v_or_b32_e32 v0, v1, v0 +; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_s_bf16_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: s_movk_i32 s4, 0x7fff +; GFX8-NEXT: v_mov_b32_e32 v1, s16 +; GFX8-NEXT: v_bfi_b32 v0, s4, v1, v0 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_s_bf16_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: s_movk_i32 s4, 0x7fff +; GFX9-NEXT: v_mov_b32_e32 v1, s16 +; GFX9-NEXT: v_bfi_b32 v0, s4, v1, v0 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_s_bf16_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s16, v0 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_s_bf16_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + ret bfloat %op +} + +define bfloat @v_copysign_bf16_f32(bfloat %mag, float %sign.f32) { +; GCN-LABEL: v_copysign_bf16_f32: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GCN-NEXT: v_and_b32_e32 v1, 0x80000000, v1 +; GCN-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, v0, v1 +; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_bf16_f32: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GFX7-NEXT: v_and_b32_e32 v1, 0x80000000, v1 +; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GFX7-NEXT: v_or_b32_e32 v0, v0, v1 +; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_bf16_f32: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX8-NEXT: s_movk_i32 s4, 0x7fff +; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_bf16_f32: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX9-NEXT: s_movk_i32 s4, 0x7fff +; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_bf16_f32: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_bf16_f32: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %sign = fptrunc float %sign.f32 to bfloat + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + ret bfloat %op +} + +define bfloat @v_copysign_bf16_f64(bfloat %mag, double %sign.f64) { +; GCN-LABEL: v_copysign_bf16_f64: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GCN-NEXT: v_and_b32_e32 v1, 0x80000000, v2 +; GCN-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, v0, v1 +; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_bf16_f64: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GFX7-NEXT: v_and_b32_e32 v1, 0x80000000, v2 +; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GFX7-NEXT: v_or_b32_e32 v0, v0, v1 +; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_bf16_f64: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: v_lshrrev_b32_e32 v1, 16, v2 +; GFX8-NEXT: s_movk_i32 s4, 0x7fff +; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_bf16_f64: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: v_lshrrev_b32_e32 v1, 16, v2 +; GFX9-NEXT: s_movk_i32 s4, 0x7fff +; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_bf16_f64: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_lshrrev_b32_e32 v1, 16, v2 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_bf16_f64: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_lshrrev_b32_e32 v1, 16, v2 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %sign = fptrunc double %sign.f64 to bfloat + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + ret bfloat %op +} + +define bfloat @v_copysign_bf16_f16(bfloat %mag, half %sign.f16) { +; GCN-LABEL: v_copysign_bf16_f16: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GCN-NEXT: v_cvt_f16_f32_e32 v1, v1 +; GCN-NEXT: v_and_b32_e32 v1, 0x8000, v1 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, v0, v1 +; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_bf16_f16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: v_cvt_f16_f32_e32 v1, v1 +; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 +; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GFX7-NEXT: v_and_b32_e32 v1, 0x8000, v1 +; GFX7-NEXT: v_or_b32_e32 v0, v0, v1 +; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_bf16_f16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: s_movk_i32 s4, 0x7fff +; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_bf16_f16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: s_movk_i32 s4, 0x7fff +; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_bf16_f16: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_bf16_f16: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %sign = bitcast half %sign.f16 to bfloat + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + ret bfloat %op +} + +define amdgpu_ps i32 @s_copysign_bf16_bf16(bfloat inreg %mag, bfloat inreg %sign) { +; GCN-LABEL: s_copysign_bf16_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s0 +; GCN-NEXT: s_and_b32 s0, s1, 0x80000000 +; GCN-NEXT: s_lshr_b32 s0, s0, 16 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, s0, v0 +; GCN-NEXT: v_readfirstlane_b32 s0, v0 +; GCN-NEXT: ; return to shader part epilog +; +; GFX7-LABEL: s_copysign_bf16_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: v_mul_f32_e64 v0, 1.0, s0 +; GFX7-NEXT: s_and_b32 s0, s1, 0x80000000 +; GFX7-NEXT: s_lshr_b32 s0, s0, 16 +; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GFX7-NEXT: v_or_b32_e32 v0, s0, v0 +; GFX7-NEXT: v_readfirstlane_b32 s0, v0 +; GFX7-NEXT: ; return to shader part epilog +; +; GFX8-LABEL: s_copysign_bf16_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_movk_i32 s2, 0x7fff +; GFX8-NEXT: v_mov_b32_e32 v0, s0 +; GFX8-NEXT: v_mov_b32_e32 v1, s1 +; GFX8-NEXT: v_bfi_b32 v0, s2, v0, v1 +; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX8-NEXT: v_readfirstlane_b32 s0, v0 +; GFX8-NEXT: ; return to shader part epilog +; +; GFX9-LABEL: s_copysign_bf16_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_movk_i32 s2, 0x7fff +; GFX9-NEXT: v_mov_b32_e32 v0, s0 +; GFX9-NEXT: v_mov_b32_e32 v1, s1 +; GFX9-NEXT: v_bfi_b32 v0, s2, v0, v1 +; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX9-NEXT: v_readfirstlane_b32 s0, v0 +; GFX9-NEXT: ; return to shader part epilog +; +; GFX10-LABEL: s_copysign_bf16_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: v_mov_b32_e32 v0, s1 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX10-NEXT: v_readfirstlane_b32 s0, v0 +; GFX10-NEXT: ; return to shader part epilog +; +; GFX11-LABEL: s_copysign_bf16_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: v_mov_b32_e32 v0, s1 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_readfirstlane_b32 s0, v0 +; GFX11-NEXT: ; return to shader part epilog + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + %cast = bitcast bfloat %op to i16 + %zext = zext i16 %cast to i32 + %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) + ret i32 %readlane +} + +define amdgpu_ps i32 @s_copysign_bf16_f32(bfloat inreg %mag, float inreg %sign.f32) { +; GCN-LABEL: s_copysign_bf16_f32: +; GCN: ; %bb.0: +; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s0 +; GCN-NEXT: s_and_b32 s0, s1, 0x80000000 +; GCN-NEXT: s_lshr_b32 s0, s0, 16 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, s0, v0 +; GCN-NEXT: v_readfirstlane_b32 s0, v0 +; GCN-NEXT: ; return to shader part epilog +; +; GFX7-LABEL: s_copysign_bf16_f32: +; GFX7: ; %bb.0: +; GFX7-NEXT: v_mul_f32_e64 v0, 1.0, s0 +; GFX7-NEXT: s_and_b32 s0, s1, 0x80000000 +; GFX7-NEXT: s_lshr_b32 s0, s0, 16 +; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GFX7-NEXT: v_or_b32_e32 v0, s0, v0 +; GFX7-NEXT: v_readfirstlane_b32 s0, v0 +; GFX7-NEXT: ; return to shader part epilog +; +; GFX8-LABEL: s_copysign_bf16_f32: +; GFX8: ; %bb.0: +; GFX8-NEXT: v_lshrrev_b32_e64 v0, 16, s1 +; GFX8-NEXT: s_movk_i32 s1, 0x7fff +; GFX8-NEXT: v_mov_b32_e32 v1, s0 +; GFX8-NEXT: v_bfi_b32 v0, s1, v1, v0 +; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX8-NEXT: v_readfirstlane_b32 s0, v0 +; GFX8-NEXT: ; return to shader part epilog +; +; GFX9-LABEL: s_copysign_bf16_f32: +; GFX9: ; %bb.0: +; GFX9-NEXT: v_lshrrev_b32_e64 v0, 16, s1 +; GFX9-NEXT: s_movk_i32 s1, 0x7fff +; GFX9-NEXT: v_mov_b32_e32 v1, s0 +; GFX9-NEXT: v_bfi_b32 v0, s1, v1, v0 +; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX9-NEXT: v_readfirstlane_b32 s0, v0 +; GFX9-NEXT: ; return to shader part epilog +; +; GFX10-LABEL: s_copysign_bf16_f32: +; GFX10: ; %bb.0: +; GFX10-NEXT: v_lshrrev_b32_e64 v0, 16, s1 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX10-NEXT: v_readfirstlane_b32 s0, v0 +; GFX10-NEXT: ; return to shader part epilog +; +; GFX11-LABEL: s_copysign_bf16_f32: +; GFX11: ; %bb.0: +; GFX11-NEXT: v_lshrrev_b32_e64 v0, 16, s1 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_readfirstlane_b32 s0, v0 +; GFX11-NEXT: ; return to shader part epilog + %sign = fptrunc float %sign.f32 to bfloat + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + %cast = bitcast bfloat %op to i16 + %zext = zext i16 %cast to i32 + %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) + ret i32 %readlane +} + +define amdgpu_ps i32 @s_copysign_bf16_f64(bfloat inreg %mag, double inreg %sign.f64) { +; GCN-LABEL: s_copysign_bf16_f64: +; GCN: ; %bb.0: +; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s0 +; GCN-NEXT: s_and_b32 s0, s2, 0x80000000 +; GCN-NEXT: s_lshr_b32 s0, s0, 16 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, s0, v0 +; GCN-NEXT: v_readfirstlane_b32 s0, v0 +; GCN-NEXT: ; return to shader part epilog +; +; GFX7-LABEL: s_copysign_bf16_f64: +; GFX7: ; %bb.0: +; GFX7-NEXT: v_mul_f32_e64 v0, 1.0, s0 +; GFX7-NEXT: s_and_b32 s0, s2, 0x80000000 +; GFX7-NEXT: s_lshr_b32 s0, s0, 16 +; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GFX7-NEXT: v_or_b32_e32 v0, s0, v0 +; GFX7-NEXT: v_readfirstlane_b32 s0, v0 +; GFX7-NEXT: ; return to shader part epilog +; +; GFX8-LABEL: s_copysign_bf16_f64: +; GFX8: ; %bb.0: +; GFX8-NEXT: v_lshrrev_b32_e64 v0, 16, s2 +; GFX8-NEXT: s_movk_i32 s1, 0x7fff +; GFX8-NEXT: v_mov_b32_e32 v1, s0 +; GFX8-NEXT: v_bfi_b32 v0, s1, v1, v0 +; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX8-NEXT: v_readfirstlane_b32 s0, v0 +; GFX8-NEXT: ; return to shader part epilog +; +; GFX9-LABEL: s_copysign_bf16_f64: +; GFX9: ; %bb.0: +; GFX9-NEXT: v_lshrrev_b32_e64 v0, 16, s2 +; GFX9-NEXT: s_movk_i32 s1, 0x7fff +; GFX9-NEXT: v_mov_b32_e32 v1, s0 +; GFX9-NEXT: v_bfi_b32 v0, s1, v1, v0 +; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX9-NEXT: v_readfirstlane_b32 s0, v0 +; GFX9-NEXT: ; return to shader part epilog +; +; GFX10-LABEL: s_copysign_bf16_f64: +; GFX10: ; %bb.0: +; GFX10-NEXT: v_lshrrev_b32_e64 v0, 16, s2 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX10-NEXT: v_readfirstlane_b32 s0, v0 +; GFX10-NEXT: ; return to shader part epilog +; +; GFX11-LABEL: s_copysign_bf16_f64: +; GFX11: ; %bb.0: +; GFX11-NEXT: v_lshrrev_b32_e64 v0, 16, s2 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_readfirstlane_b32 s0, v0 +; GFX11-NEXT: ; return to shader part epilog + %sign = fptrunc double %sign.f64 to bfloat + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + %cast = bitcast bfloat %op to i16 + %zext = zext i16 %cast to i32 + %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) + ret i32 %readlane +} + +define amdgpu_ps i32 @s_copysign_bf16_f16(bfloat inreg %mag, half inreg %sign.f16) { +; GCN-LABEL: s_copysign_bf16_f16: +; GCN: ; %bb.0: +; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s0 +; GCN-NEXT: v_cvt_f16_f32_e32 v1, s1 +; GCN-NEXT: v_and_b32_e32 v1, 0x8000, v1 +; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 +; GCN-NEXT: v_or_b32_e32 v0, v0, v1 +; GCN-NEXT: v_readfirstlane_b32 s0, v0 +; GCN-NEXT: ; return to shader part epilog +; +; GFX7-LABEL: s_copysign_bf16_f16: +; GFX7: ; %bb.0: +; GFX7-NEXT: v_cvt_f16_f32_e32 v0, s1 +; GFX7-NEXT: v_mul_f32_e64 v1, 1.0, s0 +; GFX7-NEXT: v_bfe_u32 v1, v1, 16, 15 +; GFX7-NEXT: v_and_b32_e32 v0, 0x8000, v0 +; GFX7-NEXT: v_or_b32_e32 v0, v1, v0 +; GFX7-NEXT: v_readfirstlane_b32 s0, v0 +; GFX7-NEXT: ; return to shader part epilog +; +; GFX8-LABEL: s_copysign_bf16_f16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_movk_i32 s2, 0x7fff +; GFX8-NEXT: v_mov_b32_e32 v0, s0 +; GFX8-NEXT: v_mov_b32_e32 v1, s1 +; GFX8-NEXT: v_bfi_b32 v0, s2, v0, v1 +; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX8-NEXT: v_readfirstlane_b32 s0, v0 +; GFX8-NEXT: ; return to shader part epilog +; +; GFX9-LABEL: s_copysign_bf16_f16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_movk_i32 s2, 0x7fff +; GFX9-NEXT: v_mov_b32_e32 v0, s0 +; GFX9-NEXT: v_mov_b32_e32 v1, s1 +; GFX9-NEXT: v_bfi_b32 v0, s2, v0, v1 +; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX9-NEXT: v_readfirstlane_b32 s0, v0 +; GFX9-NEXT: ; return to shader part epilog +; +; GFX10-LABEL: s_copysign_bf16_f16: +; GFX10: ; %bb.0: +; GFX10-NEXT: v_mov_b32_e32 v0, s1 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX10-NEXT: v_readfirstlane_b32 s0, v0 +; GFX10-NEXT: ; return to shader part epilog +; +; GFX11-LABEL: s_copysign_bf16_f16: +; GFX11: ; %bb.0: +; GFX11-NEXT: v_mov_b32_e32 v0, s1 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_readfirstlane_b32 s0, v0 +; GFX11-NEXT: ; return to shader part epilog + %sign = bitcast half %sign.f16 to bfloat + %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) + %cast = bitcast bfloat %op to i16 + %zext = zext i16 %cast to i32 + %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) + ret i32 %readlane +} + +declare float @llvm.copysign.f32(float, float) + +define float @v_copysign_f32_bf16(float %mag, bfloat %sign.bf16) { +; GCN-LABEL: v_copysign_f32_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: s_brev_b32 s4, -2 +; GCN-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_f32_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: s_brev_b32 s4, -2 +; GFX7-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_f32_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: v_lshlrev_b32_e32 v1, 16, v1 +; GFX8-NEXT: s_brev_b32 s4, -2 +; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_f32_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: v_lshlrev_b32_e32 v1, 16, v1 +; GFX9-NEXT: s_brev_b32 s4, -2 +; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_f32_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_lshlrev_b32_e32 v1, 16, v1 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fffffff, v0, v1 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_f32_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_lshlrev_b32_e32 v1, 16, v1 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fffffff, v0, v1 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %sign = fpext bfloat %sign.bf16 to float + %op = call float @llvm.copysign.f32(float %mag, float %sign) + ret float %op +} + +define amdgpu_ps i32 @s_copysign_f32_bf16(float inreg %mag, bfloat inreg %sign.bf16) { +; GCN-LABEL: s_copysign_f32_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: s_brev_b32 s2, -2 +; GCN-NEXT: v_mov_b32_e32 v0, s0 +; GCN-NEXT: v_mov_b32_e32 v1, s1 +; GCN-NEXT: v_bfi_b32 v0, s2, v0, v1 +; GCN-NEXT: v_readfirstlane_b32 s0, v0 +; GCN-NEXT: ; return to shader part epilog +; +; GFX7-LABEL: s_copysign_f32_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_brev_b32 s2, -2 +; GFX7-NEXT: v_mov_b32_e32 v0, s0 +; GFX7-NEXT: v_mov_b32_e32 v1, s1 +; GFX7-NEXT: v_bfi_b32 v0, s2, v0, v1 +; GFX7-NEXT: v_readfirstlane_b32 s0, v0 +; GFX7-NEXT: ; return to shader part epilog +; +; GFX8-LABEL: s_copysign_f32_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: v_lshlrev_b32_e64 v0, 16, s1 +; GFX8-NEXT: s_brev_b32 s1, -2 +; GFX8-NEXT: v_mov_b32_e32 v1, s0 +; GFX8-NEXT: v_bfi_b32 v0, s1, v1, v0 +; GFX8-NEXT: v_readfirstlane_b32 s0, v0 +; GFX8-NEXT: ; return to shader part epilog +; +; GFX9-LABEL: s_copysign_f32_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: v_lshlrev_b32_e64 v0, 16, s1 +; GFX9-NEXT: s_brev_b32 s1, -2 +; GFX9-NEXT: v_mov_b32_e32 v1, s0 +; GFX9-NEXT: v_bfi_b32 v0, s1, v1, v0 +; GFX9-NEXT: v_readfirstlane_b32 s0, v0 +; GFX9-NEXT: ; return to shader part epilog +; +; GFX10-LABEL: s_copysign_f32_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: v_lshlrev_b32_e64 v0, 16, s1 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fffffff, s0, v0 +; GFX10-NEXT: v_readfirstlane_b32 s0, v0 +; GFX10-NEXT: ; return to shader part epilog +; +; GFX11-LABEL: s_copysign_f32_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: v_lshlrev_b32_e64 v0, 16, s1 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fffffff, s0, v0 +; GFX11-NEXT: v_readfirstlane_b32 s0, v0 +; GFX11-NEXT: ; return to shader part epilog + %sign = fpext bfloat %sign.bf16 to float + %op = call float @llvm.copysign.f32(float %mag, float %sign) + %cast = bitcast float %op to i32 + %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %cast) + ret i32 %readlane +} + +declare half @llvm.copysign.f16(half, half) + +define half @v_copysign_f16_bf16(half %mag, bfloat %sign.bf16) { +; GCN-LABEL: v_copysign_f16_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: v_mul_f32_e32 v1, 1.0, v1 +; GCN-NEXT: v_cvt_f16_f32_e32 v0, v0 +; GCN-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GCN-NEXT: v_cvt_f32_f16_e32 v0, v0 +; GCN-NEXT: v_cvt_f32_f16_e32 v1, v1 +; GCN-NEXT: s_brev_b32 s4, -2 +; GCN-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_f16_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: v_cvt_f16_f32_e32 v0, v0 +; GFX7-NEXT: v_mul_f32_e32 v1, 1.0, v1 +; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX7-NEXT: v_cvt_f32_f16_e32 v1, v1 +; GFX7-NEXT: v_cvt_f32_f16_e32 v0, v0 +; GFX7-NEXT: s_brev_b32 s4, -2 +; GFX7-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_f16_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: s_movk_i32 s4, 0x7fff +; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_f16_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: s_movk_i32 s4, 0x7fff +; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_f16_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_f16_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %sign = bitcast bfloat %sign.bf16 to half + %op = call half @llvm.copysign.f16(half %mag, half %sign) + ret half %op +} + +define amdgpu_ps i32 @s_copysign_f16_bf16(half inreg %mag, bfloat inreg %sign.bf16) { +; GCN-LABEL: s_copysign_f16_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s1 +; GCN-NEXT: v_cvt_f16_f32_e32 v1, s0 +; GCN-NEXT: v_lshrrev_b32_e32 v0, 16, v0 +; GCN-NEXT: v_cvt_f32_f16_e32 v1, v1 +; GCN-NEXT: v_cvt_f32_f16_e32 v0, v0 +; GCN-NEXT: s_brev_b32 s0, -2 +; GCN-NEXT: v_bfi_b32 v0, s0, v1, v0 +; GCN-NEXT: v_cvt_f16_f32_e32 v0, v0 +; GCN-NEXT: v_readfirstlane_b32 s0, v0 +; GCN-NEXT: ; return to shader part epilog +; +; GFX7-LABEL: s_copysign_f16_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: v_cvt_f16_f32_e32 v0, s0 +; GFX7-NEXT: v_mul_f32_e64 v1, 1.0, s1 +; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 +; GFX7-NEXT: v_cvt_f32_f16_e32 v1, v1 +; GFX7-NEXT: v_cvt_f32_f16_e32 v0, v0 +; GFX7-NEXT: s_brev_b32 s0, -2 +; GFX7-NEXT: v_bfi_b32 v0, s0, v0, v1 +; GFX7-NEXT: v_cvt_f16_f32_e32 v0, v0 +; GFX7-NEXT: v_readfirstlane_b32 s0, v0 +; GFX7-NEXT: ; return to shader part epilog +; +; GFX8-LABEL: s_copysign_f16_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_movk_i32 s2, 0x7fff +; GFX8-NEXT: v_mov_b32_e32 v0, s0 +; GFX8-NEXT: v_mov_b32_e32 v1, s1 +; GFX8-NEXT: v_bfi_b32 v0, s2, v0, v1 +; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX8-NEXT: v_readfirstlane_b32 s0, v0 +; GFX8-NEXT: ; return to shader part epilog +; +; GFX9-LABEL: s_copysign_f16_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_movk_i32 s2, 0x7fff +; GFX9-NEXT: v_mov_b32_e32 v0, s0 +; GFX9-NEXT: v_mov_b32_e32 v1, s1 +; GFX9-NEXT: v_bfi_b32 v0, s2, v0, v1 +; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX9-NEXT: v_readfirstlane_b32 s0, v0 +; GFX9-NEXT: ; return to shader part epilog +; +; GFX10-LABEL: s_copysign_f16_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: v_mov_b32_e32 v0, s1 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX10-NEXT: v_readfirstlane_b32 s0, v0 +; GFX10-NEXT: ; return to shader part epilog +; +; GFX11-LABEL: s_copysign_f16_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: v_mov_b32_e32 v0, s1 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 +; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_readfirstlane_b32 s0, v0 +; GFX11-NEXT: ; return to shader part epilog + %sign = bitcast bfloat %sign.bf16 to half + %op = call half @llvm.copysign.f16(half %mag, half %sign) + %cast = bitcast half %op to i16 + %zext = zext i16 %cast to i32 + %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) + ret i32 %readlane +} + +declare double @llvm.copysign.f64(double, double) + +define double @v_copysign_f64_bf16(double %mag, bfloat %sign.bf16) { +; GCN-LABEL: v_copysign_f64_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GCN-NEXT: s_brev_b32 s4, -2 +; GCN-NEXT: v_bfi_b32 v1, s4, v1, v2 +; GCN-NEXT: s_setpc_b64 s[30:31] +; +; GFX7-LABEL: v_copysign_f64_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX7-NEXT: s_brev_b32 s4, -2 +; GFX7-NEXT: v_bfi_b32 v1, s4, v1, v2 +; GFX7-NEXT: s_setpc_b64 s[30:31] +; +; GFX8-LABEL: v_copysign_f64_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX8-NEXT: v_lshlrev_b32_e32 v2, 16, v2 +; GFX8-NEXT: s_brev_b32 s4, -2 +; GFX8-NEXT: v_bfi_b32 v1, s4, v1, v2 +; GFX8-NEXT: s_setpc_b64 s[30:31] +; +; GFX9-LABEL: v_copysign_f64_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX9-NEXT: v_lshlrev_b32_e32 v2, 16, v2 +; GFX9-NEXT: s_brev_b32 s4, -2 +; GFX9-NEXT: v_bfi_b32 v1, s4, v1, v2 +; GFX9-NEXT: s_setpc_b64 s[30:31] +; +; GFX10-LABEL: v_copysign_f64_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX10-NEXT: v_lshlrev_b32_e32 v2, 16, v2 +; GFX10-NEXT: v_bfi_b32 v1, 0x7fffffff, v1, v2 +; GFX10-NEXT: s_setpc_b64 s[30:31] +; +; GFX11-LABEL: v_copysign_f64_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) +; GFX11-NEXT: v_lshlrev_b32_e32 v2, 16, v2 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v1, 0x7fffffff, v1, v2 +; GFX11-NEXT: s_setpc_b64 s[30:31] + %sign = fpext bfloat %sign.bf16 to double + %op = call double @llvm.copysign.f64(double %mag, double %sign) + ret double %op +} + +define amdgpu_ps <2 x i32> @s_copysign_f64_bf16(double inreg %mag, bfloat inreg %sign.bf16) { +; GCN-LABEL: s_copysign_f64_bf16: +; GCN: ; %bb.0: +; GCN-NEXT: s_brev_b32 s3, -2 +; GCN-NEXT: v_mov_b32_e32 v0, s1 +; GCN-NEXT: v_mov_b32_e32 v1, s2 +; GCN-NEXT: v_bfi_b32 v0, s3, v0, v1 +; GCN-NEXT: v_readfirstlane_b32 s1, v0 +; GCN-NEXT: ; return to shader part epilog +; +; GFX7-LABEL: s_copysign_f64_bf16: +; GFX7: ; %bb.0: +; GFX7-NEXT: s_brev_b32 s3, -2 +; GFX7-NEXT: v_mov_b32_e32 v0, s1 +; GFX7-NEXT: v_mov_b32_e32 v1, s2 +; GFX7-NEXT: v_bfi_b32 v0, s3, v0, v1 +; GFX7-NEXT: v_readfirstlane_b32 s1, v0 +; GFX7-NEXT: ; return to shader part epilog +; +; GFX8-LABEL: s_copysign_f64_bf16: +; GFX8: ; %bb.0: +; GFX8-NEXT: v_lshlrev_b32_e64 v0, 16, s2 +; GFX8-NEXT: s_brev_b32 s2, -2 +; GFX8-NEXT: v_mov_b32_e32 v1, s1 +; GFX8-NEXT: v_bfi_b32 v0, s2, v1, v0 +; GFX8-NEXT: v_readfirstlane_b32 s1, v0 +; GFX8-NEXT: ; return to shader part epilog +; +; GFX9-LABEL: s_copysign_f64_bf16: +; GFX9: ; %bb.0: +; GFX9-NEXT: v_lshlrev_b32_e64 v0, 16, s2 +; GFX9-NEXT: s_brev_b32 s2, -2 +; GFX9-NEXT: v_mov_b32_e32 v1, s1 +; GFX9-NEXT: v_bfi_b32 v0, s2, v1, v0 +; GFX9-NEXT: v_readfirstlane_b32 s1, v0 +; GFX9-NEXT: ; return to shader part epilog +; +; GFX10-LABEL: s_copysign_f64_bf16: +; GFX10: ; %bb.0: +; GFX10-NEXT: v_lshlrev_b32_e64 v0, 16, s2 +; GFX10-NEXT: v_bfi_b32 v0, 0x7fffffff, s1, v0 +; GFX10-NEXT: v_readfirstlane_b32 s1, v0 +; GFX10-NEXT: ; return to shader part epilog +; +; GFX11-LABEL: s_copysign_f64_bf16: +; GFX11: ; %bb.0: +; GFX11-NEXT: v_lshlrev_b32_e64 v0, 16, s2 +; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) +; GFX11-NEXT: v_bfi_b32 v0, 0x7fffffff, s1, v0 +; GFX11-NEXT: v_readfirstlane_b32 s1, v0 +; GFX11-NEXT: ; return to shader part epilog + %sign = fpext bfloat %sign.bf16 to double + %op = call double @llvm.copysign.f64(double %mag, double %sign) + %cast = bitcast double %op to <2 x i32> + %cast.0 = extractelement <2 x i32> %cast, i32 0 + %cast.1 = extractelement <2 x i32> %cast, i32 1 + %readlane0 = call i32 @llvm.amdgcn.readfirstlane(i32 %cast.0) + %readlane1 = call i32 @llvm.amdgcn.readfirstlane(i32 %cast.1) + %ins.0 = insertelement <2 x i32> poison, i32 %readlane0, i32 0 + %ins.1 = insertelement <2 x i32> %ins.0, i32 %readlane1, i32 1 + ret <2 x i32> %ins.1 +} +;; NOTE: These prefixes are unused and the list is autogenerated. Do not add tests below this line: +; GFX11FAKE16: {{.*}} +; GFX11TRUE16: {{.*}} >From 50691935c0d6b865ac40f9e0a9fccb75097479f6 Mon Sep 17 00:00:00 2001 From: Matt Arsenault <matthew.arsena...@amd.com> Date: Fri, 30 May 2025 17:58:09 +0200 Subject: [PATCH 2/2] Delete old tests --- llvm/test/CodeGen/AMDGPU/bf16.ll | 948 ------------------------------- 1 file changed, 948 deletions(-) diff --git a/llvm/test/CodeGen/AMDGPU/bf16.ll b/llvm/test/CodeGen/AMDGPU/bf16.ll index ff10e7d499f9b..516a14edbc260 100644 --- a/llvm/test/CodeGen/AMDGPU/bf16.ll +++ b/llvm/test/CodeGen/AMDGPU/bf16.ll @@ -30725,954 +30725,6 @@ define i1 @v_fcmp_true_bf16(bfloat %a, bfloat %b) { ret i1 %op } -declare bfloat @llvm.copysign.bf16(bfloat, bfloat) - -define bfloat @v_copysign_bf16_bf16(bfloat %mag, bfloat %sign) { -; GCN-LABEL: v_copysign_bf16_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GCN-NEXT: v_and_b32_e32 v1, 0x80000000, v1 -; GCN-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, v0, v1 -; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_bf16_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GFX7-NEXT: v_and_b32_e32 v1, 0x80000000, v1 -; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GFX7-NEXT: v_or_b32_e32 v0, v0, v1 -; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_bf16_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: s_movk_i32 s4, 0x7fff -; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_bf16_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: s_movk_i32 s4, 0x7fff -; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_bf16_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_bf16_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - ret bfloat %op -} - -define bfloat @v_copysign_bf16_s_bf16(bfloat %mag, bfloat inreg %sign) { -; GCN-LABEL: v_copysign_bf16_s_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GCN-NEXT: s_and_b32 s4, s16, 0x80000000 -; GCN-NEXT: s_lshr_b32 s4, s4, 16 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, s4, v0 -; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_bf16_s_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GFX7-NEXT: s_and_b32 s4, s16, 0x80000000 -; GFX7-NEXT: s_lshr_b32 s4, s4, 16 -; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GFX7-NEXT: v_or_b32_e32 v0, s4, v0 -; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_bf16_s_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: s_movk_i32 s4, 0x7fff -; GFX8-NEXT: v_mov_b32_e32 v1, s16 -; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_bf16_s_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: s_movk_i32 s4, 0x7fff -; GFX9-NEXT: v_mov_b32_e32 v1, s16 -; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_bf16_s_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, s16 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_bf16_s_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, s0 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - ret bfloat %op -} - -define bfloat @v_copysign_s_bf16_bf16(bfloat inreg %mag, bfloat %sign) { -; GCN-LABEL: v_copysign_s_bf16_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: v_mul_f32_e64 v1, 1.0, s16 -; GCN-NEXT: v_and_b32_e32 v0, 0x80000000, v0 -; GCN-NEXT: v_lshrrev_b32_e32 v0, 16, v0 -; GCN-NEXT: v_bfe_u32 v1, v1, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, v1, v0 -; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_s_bf16_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: v_mul_f32_e64 v1, 1.0, s16 -; GFX7-NEXT: v_and_b32_e32 v0, 0x80000000, v0 -; GFX7-NEXT: v_lshrrev_b32_e32 v0, 16, v0 -; GFX7-NEXT: v_bfe_u32 v1, v1, 16, 15 -; GFX7-NEXT: v_or_b32_e32 v0, v1, v0 -; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_s_bf16_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: s_movk_i32 s4, 0x7fff -; GFX8-NEXT: v_mov_b32_e32 v1, s16 -; GFX8-NEXT: v_bfi_b32 v0, s4, v1, v0 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_s_bf16_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: s_movk_i32 s4, 0x7fff -; GFX9-NEXT: v_mov_b32_e32 v1, s16 -; GFX9-NEXT: v_bfi_b32 v0, s4, v1, v0 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_s_bf16_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s16, v0 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_s_bf16_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - ret bfloat %op -} - -define bfloat @v_copysign_bf16_f32(bfloat %mag, float %sign.f32) { -; GCN-LABEL: v_copysign_bf16_f32: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GCN-NEXT: v_and_b32_e32 v1, 0x80000000, v1 -; GCN-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, v0, v1 -; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_bf16_f32: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GFX7-NEXT: v_and_b32_e32 v1, 0x80000000, v1 -; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GFX7-NEXT: v_or_b32_e32 v0, v0, v1 -; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_bf16_f32: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX8-NEXT: s_movk_i32 s4, 0x7fff -; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_bf16_f32: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX9-NEXT: s_movk_i32 s4, 0x7fff -; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_bf16_f32: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_bf16_f32: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %sign = fptrunc float %sign.f32 to bfloat - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - ret bfloat %op -} - -define bfloat @v_copysign_bf16_f64(bfloat %mag, double %sign.f64) { -; GCN-LABEL: v_copysign_bf16_f64: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GCN-NEXT: v_and_b32_e32 v1, 0x80000000, v2 -; GCN-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, v0, v1 -; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_bf16_f64: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GFX7-NEXT: v_and_b32_e32 v1, 0x80000000, v2 -; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GFX7-NEXT: v_or_b32_e32 v0, v0, v1 -; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_bf16_f64: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: v_lshrrev_b32_e32 v1, 16, v2 -; GFX8-NEXT: s_movk_i32 s4, 0x7fff -; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_bf16_f64: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: v_lshrrev_b32_e32 v1, 16, v2 -; GFX9-NEXT: s_movk_i32 s4, 0x7fff -; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_bf16_f64: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_lshrrev_b32_e32 v1, 16, v2 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_bf16_f64: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_lshrrev_b32_e32 v1, 16, v2 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %sign = fptrunc double %sign.f64 to bfloat - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - ret bfloat %op -} - -define bfloat @v_copysign_bf16_f16(bfloat %mag, half %sign.f16) { -; GCN-LABEL: v_copysign_bf16_f16: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GCN-NEXT: v_cvt_f16_f32_e32 v1, v1 -; GCN-NEXT: v_and_b32_e32 v1, 0x8000, v1 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, v0, v1 -; GCN-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_bf16_f16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: v_cvt_f16_f32_e32 v1, v1 -; GFX7-NEXT: v_mul_f32_e32 v0, 1.0, v0 -; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GFX7-NEXT: v_and_b32_e32 v1, 0x8000, v1 -; GFX7-NEXT: v_or_b32_e32 v0, v0, v1 -; GFX7-NEXT: v_lshlrev_b32_e32 v0, 16, v0 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_bf16_f16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: s_movk_i32 s4, 0x7fff -; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_bf16_f16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: s_movk_i32 s4, 0x7fff -; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_bf16_f16: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_bf16_f16: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %sign = bitcast half %sign.f16 to bfloat - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - ret bfloat %op -} - -define amdgpu_ps i32 @s_copysign_bf16_bf16(bfloat inreg %mag, bfloat inreg %sign) { -; GCN-LABEL: s_copysign_bf16_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s0 -; GCN-NEXT: s_and_b32 s0, s1, 0x80000000 -; GCN-NEXT: s_lshr_b32 s0, s0, 16 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, s0, v0 -; GCN-NEXT: v_readfirstlane_b32 s0, v0 -; GCN-NEXT: ; return to shader part epilog -; -; GFX7-LABEL: s_copysign_bf16_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: v_mul_f32_e64 v0, 1.0, s0 -; GFX7-NEXT: s_and_b32 s0, s1, 0x80000000 -; GFX7-NEXT: s_lshr_b32 s0, s0, 16 -; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GFX7-NEXT: v_or_b32_e32 v0, s0, v0 -; GFX7-NEXT: v_readfirstlane_b32 s0, v0 -; GFX7-NEXT: ; return to shader part epilog -; -; GFX8-LABEL: s_copysign_bf16_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_movk_i32 s2, 0x7fff -; GFX8-NEXT: v_mov_b32_e32 v0, s0 -; GFX8-NEXT: v_mov_b32_e32 v1, s1 -; GFX8-NEXT: v_bfi_b32 v0, s2, v0, v1 -; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX8-NEXT: v_readfirstlane_b32 s0, v0 -; GFX8-NEXT: ; return to shader part epilog -; -; GFX9-LABEL: s_copysign_bf16_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_movk_i32 s2, 0x7fff -; GFX9-NEXT: v_mov_b32_e32 v0, s0 -; GFX9-NEXT: v_mov_b32_e32 v1, s1 -; GFX9-NEXT: v_bfi_b32 v0, s2, v0, v1 -; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX9-NEXT: v_readfirstlane_b32 s0, v0 -; GFX9-NEXT: ; return to shader part epilog -; -; GFX10-LABEL: s_copysign_bf16_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: v_mov_b32_e32 v0, s1 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX10-NEXT: v_readfirstlane_b32 s0, v0 -; GFX10-NEXT: ; return to shader part epilog -; -; GFX11-LABEL: s_copysign_bf16_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: v_mov_b32_e32 v0, s1 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_readfirstlane_b32 s0, v0 -; GFX11-NEXT: ; return to shader part epilog - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - %cast = bitcast bfloat %op to i16 - %zext = zext i16 %cast to i32 - %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) - ret i32 %readlane -} - -define amdgpu_ps i32 @s_copysign_bf16_f32(bfloat inreg %mag, float inreg %sign.f32) { -; GCN-LABEL: s_copysign_bf16_f32: -; GCN: ; %bb.0: -; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s0 -; GCN-NEXT: s_and_b32 s0, s1, 0x80000000 -; GCN-NEXT: s_lshr_b32 s0, s0, 16 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, s0, v0 -; GCN-NEXT: v_readfirstlane_b32 s0, v0 -; GCN-NEXT: ; return to shader part epilog -; -; GFX7-LABEL: s_copysign_bf16_f32: -; GFX7: ; %bb.0: -; GFX7-NEXT: v_mul_f32_e64 v0, 1.0, s0 -; GFX7-NEXT: s_and_b32 s0, s1, 0x80000000 -; GFX7-NEXT: s_lshr_b32 s0, s0, 16 -; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GFX7-NEXT: v_or_b32_e32 v0, s0, v0 -; GFX7-NEXT: v_readfirstlane_b32 s0, v0 -; GFX7-NEXT: ; return to shader part epilog -; -; GFX8-LABEL: s_copysign_bf16_f32: -; GFX8: ; %bb.0: -; GFX8-NEXT: v_lshrrev_b32_e64 v0, 16, s1 -; GFX8-NEXT: s_movk_i32 s1, 0x7fff -; GFX8-NEXT: v_mov_b32_e32 v1, s0 -; GFX8-NEXT: v_bfi_b32 v0, s1, v1, v0 -; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX8-NEXT: v_readfirstlane_b32 s0, v0 -; GFX8-NEXT: ; return to shader part epilog -; -; GFX9-LABEL: s_copysign_bf16_f32: -; GFX9: ; %bb.0: -; GFX9-NEXT: v_lshrrev_b32_e64 v0, 16, s1 -; GFX9-NEXT: s_movk_i32 s1, 0x7fff -; GFX9-NEXT: v_mov_b32_e32 v1, s0 -; GFX9-NEXT: v_bfi_b32 v0, s1, v1, v0 -; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX9-NEXT: v_readfirstlane_b32 s0, v0 -; GFX9-NEXT: ; return to shader part epilog -; -; GFX10-LABEL: s_copysign_bf16_f32: -; GFX10: ; %bb.0: -; GFX10-NEXT: v_lshrrev_b32_e64 v0, 16, s1 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX10-NEXT: v_readfirstlane_b32 s0, v0 -; GFX10-NEXT: ; return to shader part epilog -; -; GFX11-LABEL: s_copysign_bf16_f32: -; GFX11: ; %bb.0: -; GFX11-NEXT: v_lshrrev_b32_e64 v0, 16, s1 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_readfirstlane_b32 s0, v0 -; GFX11-NEXT: ; return to shader part epilog - %sign = fptrunc float %sign.f32 to bfloat - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - %cast = bitcast bfloat %op to i16 - %zext = zext i16 %cast to i32 - %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) - ret i32 %readlane -} - -define amdgpu_ps i32 @s_copysign_bf16_f64(bfloat inreg %mag, double inreg %sign.f64) { -; GCN-LABEL: s_copysign_bf16_f64: -; GCN: ; %bb.0: -; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s0 -; GCN-NEXT: s_and_b32 s0, s2, 0x80000000 -; GCN-NEXT: s_lshr_b32 s0, s0, 16 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, s0, v0 -; GCN-NEXT: v_readfirstlane_b32 s0, v0 -; GCN-NEXT: ; return to shader part epilog -; -; GFX7-LABEL: s_copysign_bf16_f64: -; GFX7: ; %bb.0: -; GFX7-NEXT: v_mul_f32_e64 v0, 1.0, s0 -; GFX7-NEXT: s_and_b32 s0, s2, 0x80000000 -; GFX7-NEXT: s_lshr_b32 s0, s0, 16 -; GFX7-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GFX7-NEXT: v_or_b32_e32 v0, s0, v0 -; GFX7-NEXT: v_readfirstlane_b32 s0, v0 -; GFX7-NEXT: ; return to shader part epilog -; -; GFX8-LABEL: s_copysign_bf16_f64: -; GFX8: ; %bb.0: -; GFX8-NEXT: v_lshrrev_b32_e64 v0, 16, s2 -; GFX8-NEXT: s_movk_i32 s1, 0x7fff -; GFX8-NEXT: v_mov_b32_e32 v1, s0 -; GFX8-NEXT: v_bfi_b32 v0, s1, v1, v0 -; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX8-NEXT: v_readfirstlane_b32 s0, v0 -; GFX8-NEXT: ; return to shader part epilog -; -; GFX9-LABEL: s_copysign_bf16_f64: -; GFX9: ; %bb.0: -; GFX9-NEXT: v_lshrrev_b32_e64 v0, 16, s2 -; GFX9-NEXT: s_movk_i32 s1, 0x7fff -; GFX9-NEXT: v_mov_b32_e32 v1, s0 -; GFX9-NEXT: v_bfi_b32 v0, s1, v1, v0 -; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX9-NEXT: v_readfirstlane_b32 s0, v0 -; GFX9-NEXT: ; return to shader part epilog -; -; GFX10-LABEL: s_copysign_bf16_f64: -; GFX10: ; %bb.0: -; GFX10-NEXT: v_lshrrev_b32_e64 v0, 16, s2 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX10-NEXT: v_readfirstlane_b32 s0, v0 -; GFX10-NEXT: ; return to shader part epilog -; -; GFX11-LABEL: s_copysign_bf16_f64: -; GFX11: ; %bb.0: -; GFX11-NEXT: v_lshrrev_b32_e64 v0, 16, s2 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_readfirstlane_b32 s0, v0 -; GFX11-NEXT: ; return to shader part epilog - %sign = fptrunc double %sign.f64 to bfloat - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - %cast = bitcast bfloat %op to i16 - %zext = zext i16 %cast to i32 - %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) - ret i32 %readlane -} - -define amdgpu_ps i32 @s_copysign_bf16_f16(bfloat inreg %mag, half inreg %sign.f16) { -; GCN-LABEL: s_copysign_bf16_f16: -; GCN: ; %bb.0: -; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s0 -; GCN-NEXT: v_cvt_f16_f32_e32 v1, s1 -; GCN-NEXT: v_and_b32_e32 v1, 0x8000, v1 -; GCN-NEXT: v_bfe_u32 v0, v0, 16, 15 -; GCN-NEXT: v_or_b32_e32 v0, v0, v1 -; GCN-NEXT: v_readfirstlane_b32 s0, v0 -; GCN-NEXT: ; return to shader part epilog -; -; GFX7-LABEL: s_copysign_bf16_f16: -; GFX7: ; %bb.0: -; GFX7-NEXT: v_cvt_f16_f32_e32 v0, s1 -; GFX7-NEXT: v_mul_f32_e64 v1, 1.0, s0 -; GFX7-NEXT: v_bfe_u32 v1, v1, 16, 15 -; GFX7-NEXT: v_and_b32_e32 v0, 0x8000, v0 -; GFX7-NEXT: v_or_b32_e32 v0, v1, v0 -; GFX7-NEXT: v_readfirstlane_b32 s0, v0 -; GFX7-NEXT: ; return to shader part epilog -; -; GFX8-LABEL: s_copysign_bf16_f16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_movk_i32 s2, 0x7fff -; GFX8-NEXT: v_mov_b32_e32 v0, s0 -; GFX8-NEXT: v_mov_b32_e32 v1, s1 -; GFX8-NEXT: v_bfi_b32 v0, s2, v0, v1 -; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX8-NEXT: v_readfirstlane_b32 s0, v0 -; GFX8-NEXT: ; return to shader part epilog -; -; GFX9-LABEL: s_copysign_bf16_f16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_movk_i32 s2, 0x7fff -; GFX9-NEXT: v_mov_b32_e32 v0, s0 -; GFX9-NEXT: v_mov_b32_e32 v1, s1 -; GFX9-NEXT: v_bfi_b32 v0, s2, v0, v1 -; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX9-NEXT: v_readfirstlane_b32 s0, v0 -; GFX9-NEXT: ; return to shader part epilog -; -; GFX10-LABEL: s_copysign_bf16_f16: -; GFX10: ; %bb.0: -; GFX10-NEXT: v_mov_b32_e32 v0, s1 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX10-NEXT: v_readfirstlane_b32 s0, v0 -; GFX10-NEXT: ; return to shader part epilog -; -; GFX11-LABEL: s_copysign_bf16_f16: -; GFX11: ; %bb.0: -; GFX11-NEXT: v_mov_b32_e32 v0, s1 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_readfirstlane_b32 s0, v0 -; GFX11-NEXT: ; return to shader part epilog - %sign = bitcast half %sign.f16 to bfloat - %op = call bfloat @llvm.copysign.bf16(bfloat %mag, bfloat %sign) - %cast = bitcast bfloat %op to i16 - %zext = zext i16 %cast to i32 - %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) - ret i32 %readlane -} - -declare float @llvm.copysign.f32(float, float) - -define float @v_copysign_f32_bf16(float %mag, bfloat %sign.bf16) { -; GCN-LABEL: v_copysign_f32_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: s_brev_b32 s4, -2 -; GCN-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_f32_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: s_brev_b32 s4, -2 -; GFX7-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_f32_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: v_lshlrev_b32_e32 v1, 16, v1 -; GFX8-NEXT: s_brev_b32 s4, -2 -; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_f32_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: v_lshlrev_b32_e32 v1, 16, v1 -; GFX9-NEXT: s_brev_b32 s4, -2 -; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_f32_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_lshlrev_b32_e32 v1, 16, v1 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fffffff, v0, v1 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_f32_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_lshlrev_b32_e32 v1, 16, v1 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fffffff, v0, v1 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %sign = fpext bfloat %sign.bf16 to float - %op = call float @llvm.copysign.f32(float %mag, float %sign) - ret float %op -} - -define amdgpu_ps i32 @s_copysign_f32_bf16(float inreg %mag, bfloat inreg %sign.bf16) { -; GCN-LABEL: s_copysign_f32_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: s_brev_b32 s2, -2 -; GCN-NEXT: v_mov_b32_e32 v0, s0 -; GCN-NEXT: v_mov_b32_e32 v1, s1 -; GCN-NEXT: v_bfi_b32 v0, s2, v0, v1 -; GCN-NEXT: v_readfirstlane_b32 s0, v0 -; GCN-NEXT: ; return to shader part epilog -; -; GFX7-LABEL: s_copysign_f32_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_brev_b32 s2, -2 -; GFX7-NEXT: v_mov_b32_e32 v0, s0 -; GFX7-NEXT: v_mov_b32_e32 v1, s1 -; GFX7-NEXT: v_bfi_b32 v0, s2, v0, v1 -; GFX7-NEXT: v_readfirstlane_b32 s0, v0 -; GFX7-NEXT: ; return to shader part epilog -; -; GFX8-LABEL: s_copysign_f32_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: v_lshlrev_b32_e64 v0, 16, s1 -; GFX8-NEXT: s_brev_b32 s1, -2 -; GFX8-NEXT: v_mov_b32_e32 v1, s0 -; GFX8-NEXT: v_bfi_b32 v0, s1, v1, v0 -; GFX8-NEXT: v_readfirstlane_b32 s0, v0 -; GFX8-NEXT: ; return to shader part epilog -; -; GFX9-LABEL: s_copysign_f32_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: v_lshlrev_b32_e64 v0, 16, s1 -; GFX9-NEXT: s_brev_b32 s1, -2 -; GFX9-NEXT: v_mov_b32_e32 v1, s0 -; GFX9-NEXT: v_bfi_b32 v0, s1, v1, v0 -; GFX9-NEXT: v_readfirstlane_b32 s0, v0 -; GFX9-NEXT: ; return to shader part epilog -; -; GFX10-LABEL: s_copysign_f32_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: v_lshlrev_b32_e64 v0, 16, s1 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fffffff, s0, v0 -; GFX10-NEXT: v_readfirstlane_b32 s0, v0 -; GFX10-NEXT: ; return to shader part epilog -; -; GFX11-LABEL: s_copysign_f32_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: v_lshlrev_b32_e64 v0, 16, s1 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fffffff, s0, v0 -; GFX11-NEXT: v_readfirstlane_b32 s0, v0 -; GFX11-NEXT: ; return to shader part epilog - %sign = fpext bfloat %sign.bf16 to float - %op = call float @llvm.copysign.f32(float %mag, float %sign) - %cast = bitcast float %op to i32 - %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %cast) - ret i32 %readlane -} - -declare half @llvm.copysign.f16(half, half) - -define half @v_copysign_f16_bf16(half %mag, bfloat %sign.bf16) { -; GCN-LABEL: v_copysign_f16_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: v_mul_f32_e32 v1, 1.0, v1 -; GCN-NEXT: v_cvt_f16_f32_e32 v0, v0 -; GCN-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GCN-NEXT: v_cvt_f32_f16_e32 v0, v0 -; GCN-NEXT: v_cvt_f32_f16_e32 v1, v1 -; GCN-NEXT: s_brev_b32 s4, -2 -; GCN-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_f16_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: v_cvt_f16_f32_e32 v0, v0 -; GFX7-NEXT: v_mul_f32_e32 v1, 1.0, v1 -; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX7-NEXT: v_cvt_f32_f16_e32 v1, v1 -; GFX7-NEXT: v_cvt_f32_f16_e32 v0, v0 -; GFX7-NEXT: s_brev_b32 s4, -2 -; GFX7-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_f16_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: s_movk_i32 s4, 0x7fff -; GFX8-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_f16_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: s_movk_i32 s4, 0x7fff -; GFX9-NEXT: v_bfi_b32 v0, s4, v0, v1 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_f16_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_f16_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, v0, v1 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %sign = bitcast bfloat %sign.bf16 to half - %op = call half @llvm.copysign.f16(half %mag, half %sign) - ret half %op -} - -define amdgpu_ps i32 @s_copysign_f16_bf16(half inreg %mag, bfloat inreg %sign.bf16) { -; GCN-LABEL: s_copysign_f16_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: v_mul_f32_e64 v0, 1.0, s1 -; GCN-NEXT: v_cvt_f16_f32_e32 v1, s0 -; GCN-NEXT: v_lshrrev_b32_e32 v0, 16, v0 -; GCN-NEXT: v_cvt_f32_f16_e32 v1, v1 -; GCN-NEXT: v_cvt_f32_f16_e32 v0, v0 -; GCN-NEXT: s_brev_b32 s0, -2 -; GCN-NEXT: v_bfi_b32 v0, s0, v1, v0 -; GCN-NEXT: v_cvt_f16_f32_e32 v0, v0 -; GCN-NEXT: v_readfirstlane_b32 s0, v0 -; GCN-NEXT: ; return to shader part epilog -; -; GFX7-LABEL: s_copysign_f16_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: v_cvt_f16_f32_e32 v0, s0 -; GFX7-NEXT: v_mul_f32_e64 v1, 1.0, s1 -; GFX7-NEXT: v_lshrrev_b32_e32 v1, 16, v1 -; GFX7-NEXT: v_cvt_f32_f16_e32 v1, v1 -; GFX7-NEXT: v_cvt_f32_f16_e32 v0, v0 -; GFX7-NEXT: s_brev_b32 s0, -2 -; GFX7-NEXT: v_bfi_b32 v0, s0, v0, v1 -; GFX7-NEXT: v_cvt_f16_f32_e32 v0, v0 -; GFX7-NEXT: v_readfirstlane_b32 s0, v0 -; GFX7-NEXT: ; return to shader part epilog -; -; GFX8-LABEL: s_copysign_f16_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_movk_i32 s2, 0x7fff -; GFX8-NEXT: v_mov_b32_e32 v0, s0 -; GFX8-NEXT: v_mov_b32_e32 v1, s1 -; GFX8-NEXT: v_bfi_b32 v0, s2, v0, v1 -; GFX8-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX8-NEXT: v_readfirstlane_b32 s0, v0 -; GFX8-NEXT: ; return to shader part epilog -; -; GFX9-LABEL: s_copysign_f16_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_movk_i32 s2, 0x7fff -; GFX9-NEXT: v_mov_b32_e32 v0, s0 -; GFX9-NEXT: v_mov_b32_e32 v1, s1 -; GFX9-NEXT: v_bfi_b32 v0, s2, v0, v1 -; GFX9-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX9-NEXT: v_readfirstlane_b32 s0, v0 -; GFX9-NEXT: ; return to shader part epilog -; -; GFX10-LABEL: s_copysign_f16_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: v_mov_b32_e32 v0, s1 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX10-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX10-NEXT: v_readfirstlane_b32 s0, v0 -; GFX10-NEXT: ; return to shader part epilog -; -; GFX11-LABEL: s_copysign_f16_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: v_mov_b32_e32 v0, s1 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fff, s0, v0 -; GFX11-NEXT: v_and_b32_e32 v0, 0xffff, v0 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_readfirstlane_b32 s0, v0 -; GFX11-NEXT: ; return to shader part epilog - %sign = bitcast bfloat %sign.bf16 to half - %op = call half @llvm.copysign.f16(half %mag, half %sign) - %cast = bitcast half %op to i16 - %zext = zext i16 %cast to i32 - %readlane = call i32 @llvm.amdgcn.readfirstlane(i32 %zext) - ret i32 %readlane -} - -declare double @llvm.copysign.f64(double, double) - -define double @v_copysign_f64_bf16(double %mag, bfloat %sign.bf16) { -; GCN-LABEL: v_copysign_f64_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GCN-NEXT: s_brev_b32 s4, -2 -; GCN-NEXT: v_bfi_b32 v1, s4, v1, v2 -; GCN-NEXT: s_setpc_b64 s[30:31] -; -; GFX7-LABEL: v_copysign_f64_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX7-NEXT: s_brev_b32 s4, -2 -; GFX7-NEXT: v_bfi_b32 v1, s4, v1, v2 -; GFX7-NEXT: s_setpc_b64 s[30:31] -; -; GFX8-LABEL: v_copysign_f64_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX8-NEXT: v_lshlrev_b32_e32 v2, 16, v2 -; GFX8-NEXT: s_brev_b32 s4, -2 -; GFX8-NEXT: v_bfi_b32 v1, s4, v1, v2 -; GFX8-NEXT: s_setpc_b64 s[30:31] -; -; GFX9-LABEL: v_copysign_f64_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX9-NEXT: v_lshlrev_b32_e32 v2, 16, v2 -; GFX9-NEXT: s_brev_b32 s4, -2 -; GFX9-NEXT: v_bfi_b32 v1, s4, v1, v2 -; GFX9-NEXT: s_setpc_b64 s[30:31] -; -; GFX10-LABEL: v_copysign_f64_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX10-NEXT: v_lshlrev_b32_e32 v2, 16, v2 -; GFX10-NEXT: v_bfi_b32 v1, 0x7fffffff, v1, v2 -; GFX10-NEXT: s_setpc_b64 s[30:31] -; -; GFX11-LABEL: v_copysign_f64_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: s_waitcnt vmcnt(0) expcnt(0) lgkmcnt(0) -; GFX11-NEXT: v_lshlrev_b32_e32 v2, 16, v2 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v1, 0x7fffffff, v1, v2 -; GFX11-NEXT: s_setpc_b64 s[30:31] - %sign = fpext bfloat %sign.bf16 to double - %op = call double @llvm.copysign.f64(double %mag, double %sign) - ret double %op -} - -define amdgpu_ps <2 x i32> @s_copysign_f64_bf16(double inreg %mag, bfloat inreg %sign.bf16) { -; GCN-LABEL: s_copysign_f64_bf16: -; GCN: ; %bb.0: -; GCN-NEXT: s_brev_b32 s3, -2 -; GCN-NEXT: v_mov_b32_e32 v0, s1 -; GCN-NEXT: v_mov_b32_e32 v1, s2 -; GCN-NEXT: v_bfi_b32 v0, s3, v0, v1 -; GCN-NEXT: v_readfirstlane_b32 s1, v0 -; GCN-NEXT: ; return to shader part epilog -; -; GFX7-LABEL: s_copysign_f64_bf16: -; GFX7: ; %bb.0: -; GFX7-NEXT: s_brev_b32 s3, -2 -; GFX7-NEXT: v_mov_b32_e32 v0, s1 -; GFX7-NEXT: v_mov_b32_e32 v1, s2 -; GFX7-NEXT: v_bfi_b32 v0, s3, v0, v1 -; GFX7-NEXT: v_readfirstlane_b32 s1, v0 -; GFX7-NEXT: ; return to shader part epilog -; -; GFX8-LABEL: s_copysign_f64_bf16: -; GFX8: ; %bb.0: -; GFX8-NEXT: v_lshlrev_b32_e64 v0, 16, s2 -; GFX8-NEXT: s_brev_b32 s2, -2 -; GFX8-NEXT: v_mov_b32_e32 v1, s1 -; GFX8-NEXT: v_bfi_b32 v0, s2, v1, v0 -; GFX8-NEXT: v_readfirstlane_b32 s1, v0 -; GFX8-NEXT: ; return to shader part epilog -; -; GFX9-LABEL: s_copysign_f64_bf16: -; GFX9: ; %bb.0: -; GFX9-NEXT: v_lshlrev_b32_e64 v0, 16, s2 -; GFX9-NEXT: s_brev_b32 s2, -2 -; GFX9-NEXT: v_mov_b32_e32 v1, s1 -; GFX9-NEXT: v_bfi_b32 v0, s2, v1, v0 -; GFX9-NEXT: v_readfirstlane_b32 s1, v0 -; GFX9-NEXT: ; return to shader part epilog -; -; GFX10-LABEL: s_copysign_f64_bf16: -; GFX10: ; %bb.0: -; GFX10-NEXT: v_lshlrev_b32_e64 v0, 16, s2 -; GFX10-NEXT: v_bfi_b32 v0, 0x7fffffff, s1, v0 -; GFX10-NEXT: v_readfirstlane_b32 s1, v0 -; GFX10-NEXT: ; return to shader part epilog -; -; GFX11-LABEL: s_copysign_f64_bf16: -; GFX11: ; %bb.0: -; GFX11-NEXT: v_lshlrev_b32_e64 v0, 16, s2 -; GFX11-NEXT: s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1) -; GFX11-NEXT: v_bfi_b32 v0, 0x7fffffff, s1, v0 -; GFX11-NEXT: v_readfirstlane_b32 s1, v0 -; GFX11-NEXT: ; return to shader part epilog - %sign = fpext bfloat %sign.bf16 to double - %op = call double @llvm.copysign.f64(double %mag, double %sign) - %cast = bitcast double %op to <2 x i32> - %cast.0 = extractelement <2 x i32> %cast, i32 0 - %cast.1 = extractelement <2 x i32> %cast, i32 1 - %readlane0 = call i32 @llvm.amdgcn.readfirstlane(i32 %cast.0) - %readlane1 = call i32 @llvm.amdgcn.readfirstlane(i32 %cast.1) - %ins.0 = insertelement <2 x i32> poison, i32 %readlane0, i32 0 - %ins.1 = insertelement <2 x i32> %ins.0, i32 %readlane1, i32 1 - ret <2 x i32> %ins.1 -} - define i16 @v_fptosi_bf16_to_i16(bfloat %x) { ; GCN-LABEL: v_fptosi_bf16_to_i16: ; GCN: ; %bb.0: _______________________________________________ llvm-branch-commits mailing list llvm-branch-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits