[llvm] AMDGPU: Add a few unsupported checks for llvm.fptrunc.round intrinsic (PR #107330)

[Changpeng Fang via llvm-commits]

https://github.com/changpeng created https://github.com/llvm/llvm-project/pull/107330

  A check here can be removed when we implement support for the corresponding types/mode.

>From 5a10479951d164f3a9805f9f3dc20f39833903c5 Mon Sep 17 00:00:00 2001
From: Changpeng Fang <changpeng.fang at amd.com>
Date: Wed, 4 Sep 2024 16:38:54 -0700
Subject: [PATCH] AMDGPU: Add a few unsupported checks for llvm.fptrunc.round
 intrinsic

  A check here can be removed when we implement support for the corresponding
types/mode.
---
 .../CodeGen/AMDGPU/llvm.fptrunc.round.err.ll  | 55 +++++++++++++++++--
 1 file changed, 49 insertions(+), 6 deletions(-)

diff --git a/llvm/test/CodeGen/AMDGPU/llvm.fptrunc.round.err.ll b/llvm/test/CodeGen/AMDGPU/llvm.fptrunc.round.err.ll
index f1d5b07e832c48..291fe00a6177bd 100644
--- a/llvm/test/CodeGen/AMDGPU/llvm.fptrunc.round.err.ll
+++ b/llvm/test/CodeGen/AMDGPU/llvm.fptrunc.round.err.ll
@@ -1,11 +1,54 @@
-; RUN: not --crash llc -mtriple=amdgcn -mcpu=gfx1030 -o /dev/null %s 2>&1 | FileCheck %s --ignore-case --check-prefix=FAIL
-; RUN: not --crash llc -global-isel -mtriple=amdgcn -mcpu=gfx1030 -o /dev/null %s 2>&1 | FileCheck %s --ignore-case --check-prefix=FAIL
+; RUN: split-file %s %t
 
-define amdgpu_gs void @test_fptrunc_round_f64(double %a, ptr addrspace(1) %out) {
-; FAIL: LLVM ERROR: Cannot select
+; RUN: not --crash llc -global-isel=0 -mtriple=amdgcn -mcpu=gfx1030 -filetype=null %t/f16-f64-err.ll 2>&1 | FileCheck --ignore-case --check-prefix=F16-F64-FAIL %s
+; RUN: not --crash llc -global-isel=1 -mtriple=amdgcn -mcpu=gfx1030 -filetype=null %t/f16-f64-err.ll 2>&1 | FileCheck --ignore-case --check-prefix=F16-F64-FAIL %s
+
+; RUN: not --crash llc -global-isel=0 -mtriple=amdgcn -mcpu=gfx1030 -filetype=null %t/f32-f64-err.ll 2>&1 | FileCheck --ignore-case --check-prefix=F32-F64-FAIL %s
+; RUN: not --crash llc -global-isel=1 -mtriple=amdgcn -mcpu=gfx1030 -filetype=null %t/f32-f64-err.ll 2>&1 | FileCheck --ignore-case --check-prefix=F32-F64-FAIL %s
+
+; TODO: check for GISEL when bfloat is supported.
+; RUN: not --crash llc -global-isel=0 -mtriple=amdgcn -mcpu=gfx1030 -filetype=null %t/bf16-f32-err.ll 2>&1 | FileCheck --ignore-case --check-prefix=BF16-F32-FAIL %s
+; RUN: not --crash llc -global-isel=0 -mtriple=amdgcn -mcpu=gfx1030 -filetype=null %t/bf16-f64-err.ll 2>&1 | FileCheck --ignore-case --check-prefix=BF16-F64-FAIL %s
+
+; RUN: not --crash llc -global-isel=0 -mtriple=amdgcn -mcpu=gfx1030 -filetype=null %t/f16-f32-tonearestaway-err.ll 2>&1 | FileCheck --ignore-case --check-prefix=TONEARESTAWAY-FAIL %s
+; RUN: not --crash llc -global-isel=1 -mtriple=amdgcn -mcpu=gfx1030 -filetype=null %t/f16-f32-tonearestaway-err.ll 2>&1 | FileCheck --ignore-case --check-prefix=TONEARESTAWAY-FAIL %s
+
+;--- f16-f64-err.ll
+define amdgpu_gs void @test_fptrunc_round_f16_f64(double %a, ptr addrspace(1) %out) {
+; F16-F64-FAIL: LLVM ERROR: Cannot select
   %res = call half @llvm.fptrunc.round.f16.f64(double %a, metadata !"round.upward")
-  store half %res, ptr addrspace(1) %out, align 4
+  store half %res, ptr addrspace(1) %out, align 2
   ret void
 }
 
-declare half @llvm.fptrunc.round.f16.f64(double, metadata)
+;--- f32-f64-err.ll
+define amdgpu_gs void @test_fptrunc_round_f32_f64(double %a, ptr addrspace(1) %out) {
+; F32-F64-FAIL: LLVM ERROR: Cannot select
+  %res = call float @llvm.fptrunc.round.f32.f64(double %a, metadata !"round.downward")
+  store float %res, ptr addrspace(1) %out, align 4
+  ret void
+}
+
+;--- bf16-f32-err.ll
+define amdgpu_gs void @test_fptrunc_round_bf16_f32(float %a, ptr addrspace(1) %out) {
+; BF16-F32-FAIL: LLVM ERROR: Cannot select
+  %res = call bfloat @llvm.fptrunc.round.bf16.f32(float %a, metadata !"round.towardzero")
+  store bfloat %res, ptr addrspace(1) %out, align 2
+  ret void
+}
+
+;--- bf16-f64-err.ll
+define amdgpu_gs void @test_fptrunc_round_bf16_f64(double %a, ptr addrspace(1) %out) {
+; BF16-F64-FAIL: LLVM ERROR: Cannot select
+  %res = call bfloat @llvm.fptrunc.round.bf16.f32(double %a, metadata !"round.tonearest")
+  store bfloat %res, ptr addrspace(1) %out, align 2
+  ret void
+}
+
+;--- f16-f32-tonearestaway-err.ll
+define amdgpu_gs void @test_fptrunc_round_f16_f32_tonearestaway(float %a, ptr addrspace(1) %out) {
+; TONEARESTAWAY-FAIL: LLVM ERROR: Cannot select
+  %res = call half @llvm.fptrunc.round.f16.f32(float %a, metadata !"round.tonearestaway")
+  store half %res, ptr addrspace(1) %out, align 2
+  ret void
+}