llvm/test/CodeGen/AMDGPU/llvm.log2.bf16.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
; RUN: llc -global-isel=0 -mtriple=amdgpu12.50-amd-amdhsa -mattr=+real-true16 %s -o - | FileCheck -check-prefixes=GFX-TRUE16,GFX-SDAG-TRUE16 %s
; RUN: llc -global-isel=0 -mtriple=amdgpu12.50-amd-amdhsa -mattr=-real-true16 %s -o - | FileCheck -check-prefixes=GFX-FAKE16,GFX-SDAG-FAKE16 %s
; RUN: llc -global-isel=1 -mtriple=amdgpu12.50-amd-amdhsa -mattr=+real-true16 %s -o - | FileCheck -check-prefixes=GFX-TRUE16,GFX-GISEL-TRUE16 %s
; RUN: llc -global-isel=1 -mtriple=amdgpu12.50-amd-amdhsa -mattr=-real-true16 %s -o - | FileCheck -check-prefixes=GFX-FAKE16,GFX-GISEL-FAKE16 %s

define bfloat @v_log2_bf16(bfloat %in) {
; GFX-TRUE16-LABEL: v_log2_bf16:
; GFX-TRUE16:       ; %bb.0:
; GFX-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-TRUE16-NEXT:    v_log_bf16_e32 v0.l, v0.l
; GFX-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-FAKE16-LABEL: v_log2_bf16:
; GFX-FAKE16:       ; %bb.0:
; GFX-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-FAKE16-NEXT:    v_log_bf16_e32 v0, v0
; GFX-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %result = call bfloat @llvm.log2.bf16(bfloat %in)
  ret bfloat %result
}

define bfloat @v_log2_fabs_bf16(bfloat %in) {
; GFX-TRUE16-LABEL: v_log2_fabs_bf16:
; GFX-TRUE16:       ; %bb.0:
; GFX-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-TRUE16-NEXT:    v_log_bf16_e64 v0.l, |v0.l|
; GFX-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-FAKE16-LABEL: v_log2_fabs_bf16:
; GFX-FAKE16:       ; %bb.0:
; GFX-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-FAKE16-NEXT:    v_log_bf16_e64 v0, |v0|
; GFX-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %fabs = call bfloat @llvm.fabs.bf16(bfloat %in)
  %result = call bfloat @llvm.log2.bf16(bfloat %fabs)
  ret bfloat %result
}

define bfloat @v_log2_fneg_fabs_bf16(bfloat %in) {
; GFX-TRUE16-LABEL: v_log2_fneg_fabs_bf16:
; GFX-TRUE16:       ; %bb.0:
; GFX-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-TRUE16-NEXT:    v_log_bf16_e64 v0.l, -|v0.l|
; GFX-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-FAKE16-LABEL: v_log2_fneg_fabs_bf16:
; GFX-FAKE16:       ; %bb.0:
; GFX-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-FAKE16-NEXT:    v_log_bf16_e64 v0, -|v0|
; GFX-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %fabs = call bfloat @llvm.fabs.bf16(bfloat %in)
  %fneg.fabs = fneg bfloat %fabs
  %result = call bfloat @llvm.log2.bf16(bfloat %fneg.fabs)
  ret bfloat %result
}

define bfloat @v_log2_fneg_bf16(bfloat %in) {
; GFX-TRUE16-LABEL: v_log2_fneg_bf16:
; GFX-TRUE16:       ; %bb.0:
; GFX-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-TRUE16-NEXT:    v_log_bf16_e64 v0.l, -v0.l
; GFX-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-FAKE16-LABEL: v_log2_fneg_bf16:
; GFX-FAKE16:       ; %bb.0:
; GFX-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-FAKE16-NEXT:    v_log_bf16_e64 v0, -v0
; GFX-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %fneg = fneg bfloat %in
  %result = call bfloat @llvm.log2.bf16(bfloat %fneg)
  ret bfloat %result
}

define bfloat @v_log2_bf16_fast(bfloat %in) {
; GFX-TRUE16-LABEL: v_log2_bf16_fast:
; GFX-TRUE16:       ; %bb.0:
; GFX-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-TRUE16-NEXT:    v_log_bf16_e32 v0.l, v0.l
; GFX-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-FAKE16-LABEL: v_log2_bf16_fast:
; GFX-FAKE16:       ; %bb.0:
; GFX-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-FAKE16-NEXT:    v_log_bf16_e32 v0, v0
; GFX-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %result = call fast bfloat @llvm.log2.bf16(bfloat %in)
  ret bfloat %result
}

define <2 x bfloat> @v_log2_v2bf16(<2 x bfloat> %in) {
; GFX-TRUE16-LABEL: v_log2_v2bf16:
; GFX-TRUE16:       ; %bb.0:
; GFX-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-TRUE16-NEXT:    v_log_bf16_e32 v0.l, v0.l
; GFX-TRUE16-NEXT:    v_log_bf16_e32 v0.h, v0.h
; GFX-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-SDAG-FAKE16-LABEL: v_log2_v2bf16:
; GFX-SDAG-FAKE16:       ; %bb.0:
; GFX-SDAG-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-SDAG-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e64 v1, v0 op_sel:[1,0]
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e32 v0, v0
; GFX-SDAG-FAKE16-NEXT:    v_nop
; GFX-SDAG-FAKE16-NEXT:    s_delay_alu instid0(TRANS32_DEP_1)
; GFX-SDAG-FAKE16-NEXT:    v_perm_b32 v0, v1, v0, 0x5040100
; GFX-SDAG-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-GISEL-FAKE16-LABEL: v_log2_v2bf16:
; GFX-GISEL-FAKE16:       ; %bb.0:
; GFX-GISEL-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-GISEL-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e32 v1, v0
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e64 v0, v0 op_sel:[1,0]
; GFX-GISEL-FAKE16-NEXT:    v_nop
; GFX-GISEL-FAKE16-NEXT:    s_delay_alu instid0(TRANS32_DEP_1)
; GFX-GISEL-FAKE16-NEXT:    v_pack_b32_f16 v0, v1, v0
; GFX-GISEL-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %result = call <2 x bfloat> @llvm.log2.v2bf16(<2 x bfloat> %in)
  ret <2 x bfloat> %result
}

define <2 x bfloat> @v_log2_fabs_v2bf16(<2 x bfloat> %in) {
; GFX-SDAG-TRUE16-LABEL: v_log2_fabs_v2bf16:
; GFX-SDAG-TRUE16:       ; %bb.0:
; GFX-SDAG-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-SDAG-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-SDAG-TRUE16-NEXT:    v_log_bf16_e64 v0.l, |v0.l|
; GFX-SDAG-TRUE16-NEXT:    v_log_bf16_e64 v0.h, |v0.h|
; GFX-SDAG-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-SDAG-FAKE16-LABEL: v_log2_fabs_v2bf16:
; GFX-SDAG-FAKE16:       ; %bb.0:
; GFX-SDAG-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-SDAG-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e64 v1, |v0| op_sel:[1,0]
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e64 v0, |v0|
; GFX-SDAG-FAKE16-NEXT:    v_nop
; GFX-SDAG-FAKE16-NEXT:    s_delay_alu instid0(TRANS32_DEP_1)
; GFX-SDAG-FAKE16-NEXT:    v_perm_b32 v0, v1, v0, 0x5040100
; GFX-SDAG-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-GISEL-TRUE16-LABEL: v_log2_fabs_v2bf16:
; GFX-GISEL-TRUE16:       ; %bb.0:
; GFX-GISEL-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-GISEL-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-GISEL-TRUE16-NEXT:    v_and_b32_e32 v0, 0x7fff7fff, v0
; GFX-GISEL-TRUE16-NEXT:    s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1)
; GFX-GISEL-TRUE16-NEXT:    v_log_bf16_e32 v0.l, v0.l
; GFX-GISEL-TRUE16-NEXT:    v_log_bf16_e32 v0.h, v0.h
; GFX-GISEL-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-GISEL-FAKE16-LABEL: v_log2_fabs_v2bf16:
; GFX-GISEL-FAKE16:       ; %bb.0:
; GFX-GISEL-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-GISEL-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-GISEL-FAKE16-NEXT:    v_and_b32_e32 v0, 0x7fff7fff, v0
; GFX-GISEL-FAKE16-NEXT:    s_delay_alu instid0(VALU_DEP_1) | instskip(SKIP_2) | instid1(TRANS32_DEP_1)
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e32 v1, v0
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e64 v0, v0 op_sel:[1,0]
; GFX-GISEL-FAKE16-NEXT:    v_nop
; GFX-GISEL-FAKE16-NEXT:    v_pack_b32_f16 v0, v1, v0
; GFX-GISEL-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %fabs = call <2 x bfloat> @llvm.fabs.v2bf16(<2 x bfloat> %in)
  %result = call <2 x bfloat> @llvm.log2.v2bf16(<2 x bfloat> %fabs)
  ret <2 x bfloat> %result
}

define <2 x bfloat> @v_log2_fneg_fabs_v2bf16(<2 x bfloat> %in) {
; GFX-SDAG-TRUE16-LABEL: v_log2_fneg_fabs_v2bf16:
; GFX-SDAG-TRUE16:       ; %bb.0:
; GFX-SDAG-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-SDAG-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-SDAG-TRUE16-NEXT:    v_log_bf16_e64 v0.l, -|v0.l|
; GFX-SDAG-TRUE16-NEXT:    v_log_bf16_e64 v0.h, -|v0.h|
; GFX-SDAG-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-SDAG-FAKE16-LABEL: v_log2_fneg_fabs_v2bf16:
; GFX-SDAG-FAKE16:       ; %bb.0:
; GFX-SDAG-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-SDAG-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e64 v1, -|v0| op_sel:[1,0]
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e64 v0, -|v0|
; GFX-SDAG-FAKE16-NEXT:    v_nop
; GFX-SDAG-FAKE16-NEXT:    s_delay_alu instid0(TRANS32_DEP_1)
; GFX-SDAG-FAKE16-NEXT:    v_perm_b32 v0, v1, v0, 0x5040100
; GFX-SDAG-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-GISEL-TRUE16-LABEL: v_log2_fneg_fabs_v2bf16:
; GFX-GISEL-TRUE16:       ; %bb.0:
; GFX-GISEL-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-GISEL-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-GISEL-TRUE16-NEXT:    v_or_b32_e32 v0, 0x80008000, v0
; GFX-GISEL-TRUE16-NEXT:    s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1)
; GFX-GISEL-TRUE16-NEXT:    v_log_bf16_e32 v0.l, v0.l
; GFX-GISEL-TRUE16-NEXT:    v_log_bf16_e32 v0.h, v0.h
; GFX-GISEL-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-GISEL-FAKE16-LABEL: v_log2_fneg_fabs_v2bf16:
; GFX-GISEL-FAKE16:       ; %bb.0:
; GFX-GISEL-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-GISEL-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-GISEL-FAKE16-NEXT:    v_or_b32_e32 v0, 0x80008000, v0
; GFX-GISEL-FAKE16-NEXT:    s_delay_alu instid0(VALU_DEP_1) | instskip(SKIP_2) | instid1(TRANS32_DEP_1)
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e32 v1, v0
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e64 v0, v0 op_sel:[1,0]
; GFX-GISEL-FAKE16-NEXT:    v_nop
; GFX-GISEL-FAKE16-NEXT:    v_pack_b32_f16 v0, v1, v0
; GFX-GISEL-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %fabs = call <2 x bfloat> @llvm.fabs.v2bf16(<2 x bfloat> %in)
  %fneg.fabs = fneg <2 x bfloat> %fabs
  %result = call <2 x bfloat> @llvm.log2.v2bf16(<2 x bfloat> %fneg.fabs)
  ret <2 x bfloat> %result
}

define <2 x bfloat> @v_log2_fneg_v2bf16(<2 x bfloat> %in) {
; GFX-SDAG-TRUE16-LABEL: v_log2_fneg_v2bf16:
; GFX-SDAG-TRUE16:       ; %bb.0:
; GFX-SDAG-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-SDAG-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-SDAG-TRUE16-NEXT:    v_log_bf16_e64 v0.l, -v0.l
; GFX-SDAG-TRUE16-NEXT:    v_log_bf16_e64 v0.h, -v0.h
; GFX-SDAG-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-SDAG-FAKE16-LABEL: v_log2_fneg_v2bf16:
; GFX-SDAG-FAKE16:       ; %bb.0:
; GFX-SDAG-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-SDAG-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e64 v1, -v0 op_sel:[1,0]
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e64 v0, -v0
; GFX-SDAG-FAKE16-NEXT:    v_nop
; GFX-SDAG-FAKE16-NEXT:    s_delay_alu instid0(TRANS32_DEP_1)
; GFX-SDAG-FAKE16-NEXT:    v_perm_b32 v0, v1, v0, 0x5040100
; GFX-SDAG-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-GISEL-TRUE16-LABEL: v_log2_fneg_v2bf16:
; GFX-GISEL-TRUE16:       ; %bb.0:
; GFX-GISEL-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-GISEL-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-GISEL-TRUE16-NEXT:    v_xor_b32_e32 v0, 0x80008000, v0
; GFX-GISEL-TRUE16-NEXT:    s_delay_alu instid0(VALU_DEP_1) | instskip(NEXT) | instid1(VALU_DEP_1)
; GFX-GISEL-TRUE16-NEXT:    v_log_bf16_e32 v0.l, v0.l
; GFX-GISEL-TRUE16-NEXT:    v_log_bf16_e32 v0.h, v0.h
; GFX-GISEL-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-GISEL-FAKE16-LABEL: v_log2_fneg_v2bf16:
; GFX-GISEL-FAKE16:       ; %bb.0:
; GFX-GISEL-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-GISEL-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-GISEL-FAKE16-NEXT:    v_xor_b32_e32 v0, 0x80008000, v0
; GFX-GISEL-FAKE16-NEXT:    s_delay_alu instid0(VALU_DEP_1) | instskip(SKIP_2) | instid1(TRANS32_DEP_1)
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e32 v1, v0
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e64 v0, v0 op_sel:[1,0]
; GFX-GISEL-FAKE16-NEXT:    v_nop
; GFX-GISEL-FAKE16-NEXT:    v_pack_b32_f16 v0, v1, v0
; GFX-GISEL-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %fneg = fneg <2 x bfloat> %in
  %result = call <2 x bfloat> @llvm.log2.v2bf16(<2 x bfloat> %fneg)
  ret <2 x bfloat> %result
}

define <2 x bfloat> @v_log2_v2bf16_fast(<2 x bfloat> %in) {
; GFX-TRUE16-LABEL: v_log2_v2bf16_fast:
; GFX-TRUE16:       ; %bb.0:
; GFX-TRUE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-TRUE16-NEXT:    s_wait_kmcnt 0x0
; GFX-TRUE16-NEXT:    v_log_bf16_e32 v0.l, v0.l
; GFX-TRUE16-NEXT:    v_log_bf16_e32 v0.h, v0.h
; GFX-TRUE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-SDAG-FAKE16-LABEL: v_log2_v2bf16_fast:
; GFX-SDAG-FAKE16:       ; %bb.0:
; GFX-SDAG-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-SDAG-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e64 v1, v0 op_sel:[1,0]
; GFX-SDAG-FAKE16-NEXT:    v_log_bf16_e32 v0, v0
; GFX-SDAG-FAKE16-NEXT:    v_nop
; GFX-SDAG-FAKE16-NEXT:    s_delay_alu instid0(TRANS32_DEP_1)
; GFX-SDAG-FAKE16-NEXT:    v_perm_b32 v0, v1, v0, 0x5040100
; GFX-SDAG-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
;
; GFX-GISEL-FAKE16-LABEL: v_log2_v2bf16_fast:
; GFX-GISEL-FAKE16:       ; %bb.0:
; GFX-GISEL-FAKE16-NEXT:    s_wait_loadcnt_dscnt 0x0
; GFX-GISEL-FAKE16-NEXT:    s_wait_kmcnt 0x0
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e32 v1, v0
; GFX-GISEL-FAKE16-NEXT:    v_log_bf16_e64 v0, v0 op_sel:[1,0]
; GFX-GISEL-FAKE16-NEXT:    v_nop
; GFX-GISEL-FAKE16-NEXT:    s_delay_alu instid0(TRANS32_DEP_1)
; GFX-GISEL-FAKE16-NEXT:    v_pack_b32_f16 v0, v1, v0
; GFX-GISEL-FAKE16-NEXT:    s_set_pc_i64 s[30:31]
  %result = call fast <2 x bfloat> @llvm.log2.v2bf16(<2 x bfloat> %in)
  ret <2 x bfloat> %result
}

declare bfloat @llvm.log2.bf16(bfloat) #0
declare <2 x bfloat> @llvm.log2.v2bf16(<2 x bfloat>) #0
declare bfloat @llvm.fabs.bf16(bfloat) #0
declare <2 x bfloat> @llvm.fabs.v2bf16(<2 x bfloat>) #0

attributes #0 = { nocallback nofree nosync nounwind speculatable willreturn memory(none) }