[PATCH] D147939: ValueTracking: Handle fptrunc in computeKnownFPClass

[Matt Arsenault via Phabricator via llvm-commits]

arsenm updated this revision to Diff 512970.
arsenm added a comment.

Handle infinities


CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D147939/new/

https://reviews.llvm.org/D147939

Files:
  llvm/lib/Analysis/ValueTracking.cpp
  llvm/test/Transforms/Attributor/nofpclass-fptrunc.ll


Index: llvm/test/Transforms/Attributor/nofpclass-fptrunc.ll
===================================================================
--- llvm/test/Transforms/Attributor/nofpclass-fptrunc.ll
+++ llvm/test/Transforms/Attributor/nofpclass-fptrunc.ll
@@ -12,7 +12,7 @@
 }
 
 define float @ret_fptrunc_nonan(double nofpclass(nan) %arg0) {
-; CHECK-LABEL: define float @ret_fptrunc_nonan
+; CHECK-LABEL: define nofpclass(nan) float @ret_fptrunc_nonan
 ; CHECK-SAME: (double nofpclass(nan) [[ARG0:%.*]]) #[[ATTR0]] {
 ; CHECK-NEXT:    [[EXT:%.*]] = fptrunc double [[ARG0]] to float
 ; CHECK-NEXT:    ret float [[EXT]]
@@ -42,7 +42,7 @@
 }
 
 define float @ret_fptrunc_noinf(double nofpclass(inf) %arg0) {
-; CHECK-LABEL: define float @ret_fptrunc_noinf
+; CHECK-LABEL: define nofpclass(inf) float @ret_fptrunc_noinf
 ; CHECK-SAME: (double nofpclass(inf) [[ARG0:%.*]]) #[[ATTR0]] {
 ; CHECK-NEXT:    [[EXT:%.*]] = fptrunc double [[ARG0]] to float
 ; CHECK-NEXT:    ret float [[EXT]]
@@ -52,7 +52,7 @@
 }
 
 define float @ret_fptrunc_nopinf(double nofpclass(pinf) %arg0) {
-; CHECK-LABEL: define float @ret_fptrunc_nopinf
+; CHECK-LABEL: define nofpclass(pinf) float @ret_fptrunc_nopinf
 ; CHECK-SAME: (double nofpclass(pinf) [[ARG0:%.*]]) #[[ATTR0]] {
 ; CHECK-NEXT:    [[EXT:%.*]] = fptrunc double [[ARG0]] to float
 ; CHECK-NEXT:    ret float [[EXT]]
@@ -62,7 +62,7 @@
 }
 
 define float @ret_fptrunc_noninf(double nofpclass(ninf) %arg0) {
-; CHECK-LABEL: define float @ret_fptrunc_noninf
+; CHECK-LABEL: define nofpclass(ninf) float @ret_fptrunc_noninf
 ; CHECK-SAME: (double nofpclass(ninf) [[ARG0:%.*]]) #[[ATTR0]] {
 ; CHECK-NEXT:    [[EXT:%.*]] = fptrunc double [[ARG0]] to float
 ; CHECK-NEXT:    ret float [[EXT]]
@@ -102,7 +102,7 @@
 }
 
 define float @ret_fptrunc_nonan_noinf(double nofpclass(nan inf) %arg0) {
-; CHECK-LABEL: define float @ret_fptrunc_nonan_noinf
+; CHECK-LABEL: define nofpclass(nan inf) float @ret_fptrunc_nonan_noinf
 ; CHECK-SAME: (double nofpclass(nan inf) [[ARG0:%.*]]) #[[ATTR0]] {
 ; CHECK-NEXT:    [[EXT:%.*]] = fptrunc double [[ARG0]] to float
 ; CHECK-NEXT:    ret float [[EXT]]
Index: llvm/lib/Analysis/ValueTracking.cpp
===================================================================
--- llvm/lib/Analysis/ValueTracking.cpp
+++ llvm/lib/Analysis/ValueTracking.cpp
@@ -4638,6 +4638,18 @@
       Known.SignBit = std::nullopt;
     break;
   }
+  case Instruction::FPTrunc: {
+    KnownFPClass KnownSrc;
+    computeKnownFPClass(Op->getOperand(0), DemandedElts,
+                        InterestedClasses, KnownSrc, Depth + 1, Q, TLI);
+    if (KnownSrc.isKnownNeverNaN())
+      Known.knownNot(fcNan);
+    if (KnownSrc.isKnownNeverPosInfinity())
+      Known.knownNot(fcPosInf);
+    if (KnownSrc.isKnownNeverNegInfinity())
+      Known.knownNot(fcNegInf);
+    break;
+  }
   case Instruction::SIToFP:
   case Instruction::UIToFP: {
     // Cannot produce nan


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D147939.512970.patch
Type: text/x-patch
Size: 2893 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20230412/01e620af/attachment.bin>