[llvm] r362698 - [AArch64] Handle ISD::LROUND and ISD::LLROUND for float16

[Adhemerval Zanella via llvm-commits]

Author: azanella
Date: Thu Jun  6 04:53:26 2019
New Revision: 362698

URL: http://llvm.org/viewvc/llvm-project?rev=362698&view=rev
Log:
[AArch64] Handle ISD::LROUND and ISD::LLROUND for float16

This patch is a follow up for D61391 to add lround/llround
support for float16.

Reviewed By: SjoerdMeijer

Differential Revision: https://reviews.llvm.org/D62861


Added:
    llvm/trunk/test/CodeGen/AArch64/llround-conv-fp16.ll
    llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16-win.ll
    llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16.ll
Modified:
    llvm/trunk/lib/Target/AArch64/AArch64InstrInfo.td

Modified: llvm/trunk/lib/Target/AArch64/AArch64InstrInfo.td
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/AArch64/AArch64InstrInfo.td?rev=362698&r1=362697&r2=362698&view=diff
==============================================================================
--- llvm/trunk/lib/Target/AArch64/AArch64InstrInfo.td (original)
+++ llvm/trunk/lib/Target/AArch64/AArch64InstrInfo.td Thu Jun  6 04:53:26 2019
@@ -3083,6 +3083,14 @@ defm : FPToIntegerPats<fp_to_uint, ftrun
 defm : FPToIntegerPats<fp_to_sint, fround, "FCVTAS">;
 defm : FPToIntegerPats<fp_to_uint, fround, "FCVTAU">;
 
+let Predicates = [HasFullFP16] in {
+  def : Pat<(i32 (lround f16:$Rn)),
+            (!cast<Instruction>(FCVTASUWHr) f16:$Rn)>;
+  def : Pat<(i64 (lround f16:$Rn)),
+            (!cast<Instruction>(FCVTASUXHr) f16:$Rn)>;
+  def : Pat<(i64 (llround f16:$Rn)),
+            (!cast<Instruction>(FCVTASUXHr) f16:$Rn)>;
+}
 def : Pat<(i32 (lround f32:$Rn)),
           (!cast<Instruction>(FCVTASUWSr) f32:$Rn)>;
 def : Pat<(i32 (lround f64:$Rn)),

Added: llvm/trunk/test/CodeGen/AArch64/llround-conv-fp16.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/AArch64/llround-conv-fp16.ll?rev=362698&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/AArch64/llround-conv-fp16.ll (added)
+++ llvm/trunk/test/CodeGen/AArch64/llround-conv-fp16.ll Thu Jun  6 04:53:26 2019
@@ -0,0 +1,32 @@
+; RUN: llc < %s -mtriple=aarch64 -mattr=+fullfp16 | FileCheck %s
+
+; CHECK-LABEL: testmhhs:
+; CHECK:       fcvtas  x0, h0
+; CHECK:       ret
+define i16 @testmhhs(half %x) {
+entry:
+  %0 = tail call i64 @llvm.llround.i64.f16(half %x)
+  %conv = trunc i64 %0 to i16
+  ret i16 %conv
+}
+
+; CHECK-LABEL: testmhws:
+; CHECK:       fcvtas  x0, h0
+; CHECK:       ret
+define i32 @testmhws(half %x) {
+entry:
+  %0 = tail call i64 @llvm.llround.i64.f16(half %x)
+  %conv = trunc i64 %0 to i32
+  ret i32 %conv
+}
+
+; CHECK-LABEL: testmhxs:
+; CHECK:       fcvtas  x0, h0
+; CHECK-NEXT:  ret
+define i64 @testmhxs(half %x) {
+entry:
+  %0 = tail call i64 @llvm.llround.i64.f16(half %x)
+  ret i64 %0
+}
+
+declare i64 @llvm.llround.i64.f16(half) nounwind readnone

Added: llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16-win.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16-win.ll?rev=362698&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16-win.ll (added)
+++ llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16-win.ll Thu Jun  6 04:53:26 2019
@@ -0,0 +1,33 @@
+; RUN: llc < %s -mtriple=aarch64-windows -mattr=+fullfp16 | FileCheck %s
+
+; CHECK-LABEL: testmhhs:
+; CHECK:       fcvtas  w0, h0
+; CHECK:       ret
+define i16 @testmhhs(half %x) {
+entry:
+  %0 = tail call i32 @llvm.lround.i32.f16(half %x)
+  %conv = trunc i32 %0 to i16
+  ret i16 %conv
+}
+
+; CHECK-LABEL: testmhws:
+; CHECK:       fcvtas  w0, h0
+; CHECK:       ret
+define i32 @testmhws(half %x) {
+entry:
+  %0 = tail call i32 @llvm.lround.i32.f16(half %x)
+  ret i32 %0
+}
+
+; CHECK-LABEL: testmhxs:
+; CHECK:       fcvtas  w8, h0
+; CHECK-NEXT:  sxtw    x0, w8
+; CHECK-NEXT:  ret
+define i64 @testmhxs(half %x) {
+entry:
+  %0 = tail call i32 @llvm.lround.i32.f16(half %x)
+  %conv = sext i32 %0 to i64
+  ret i64 %conv
+}
+
+declare i32 @llvm.lround.i32.f16(half) nounwind readnone

Added: llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16.ll?rev=362698&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16.ll (added)
+++ llvm/trunk/test/CodeGen/AArch64/lround-conv-fp16.ll Thu Jun  6 04:53:26 2019
@@ -0,0 +1,32 @@
+; RUN: llc < %s -mtriple=aarch64 -mattr=+fullfp16 | FileCheck %s
+
+; CHECK-LABEL: testmhhs:
+; CHECK:       fcvtas  x0, h0
+; CHECK:       ret
+define i16 @testmhhs(half %x) {
+entry:
+  %0 = tail call i64 @llvm.lround.i64.f16(half %x)
+  %conv = trunc i64 %0 to i16
+  ret i16 %conv
+}
+
+; CHECK-LABEL: testmhws:
+; CHECK:       fcvtas  x0, h0
+; CHECK:       ret
+define i32 @testmhws(half %x) {
+entry:
+  %0 = tail call i64 @llvm.lround.i64.f16(half %x)
+  %conv = trunc i64 %0 to i32
+  ret i32 %conv
+}
+
+; CHECK-LABEL: testmhxs:
+; CHECK:       fcvtas  x0, h0
+; CHECK-NEXT:  ret
+define i64 @testmhxs(half %x) {
+entry:
+  %0 = tail call i64 @llvm.lround.i64.f16(half %x)
+  ret i64 %0
+}
+
+declare i64 @llvm.lround.i64.f16(half) nounwind readnone