[PATCH] D112142: [InstCombine][ConstantFolding] Make ConstantFoldLoadThroughBitcast TypeSize-aware

[Peter Waller via Phabricator via llvm-commits]

This revision was landed with ongoing or failed builds.
This revision was automatically updated to reflect the committed changes.
Closed by commit rG98f08752f76b: [InstCombine][ConstantFolding] Make ConstantFoldLoadThroughBitcast TypeSize… (authored by peterwaller-arm).

Repository:
  rG LLVM Github Monorepo

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

https://reviews.llvm.org/D112142

Files:
  llvm/lib/Analysis/ConstantFolding.cpp
  llvm/test/Transforms/InstCombine/vscale_load.ll


Index: llvm/test/Transforms/InstCombine/vscale_load.ll
===================================================================
--- /dev/null
+++ llvm/test/Transforms/InstCombine/vscale_load.ll
@@ -0,0 +1,27 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt < %s -instcombine -S -verify | FileCheck %s
+
+define <2 x i8> @constprop_load_bitcast(<vscale x 16 x i1>* %ptr) {
+; CHECK-LABEL: @constprop_load_bitcast(
+; CHECK-NEXT:    store <vscale x 16 x i1> zeroinitializer, <vscale x 16 x i1>* [[PTR:%.*]], align 16
+; CHECK-NEXT:    ret <2 x i8> zeroinitializer
+;
+  store <vscale x 16 x i1> zeroinitializer, <vscale x 16 x i1>* %ptr, align 16
+  %cast_to_fixed = bitcast <vscale x 16 x i1>* %ptr to <2 x i8>*
+  %a = load <2 x i8>, <2 x i8>* %cast_to_fixed, align 16
+  ret <2 x i8> %a
+}
+
+; vscale-sized vec not guaranteed to fill destination.
+define <8 x i8> @constprop_load_bitcast_neg(<vscale x 16 x i1>* %ptr) {
+; CHECK-LABEL: @constprop_load_bitcast_neg(
+; CHECK-NEXT:    store <vscale x 16 x i1> zeroinitializer, <vscale x 16 x i1>* [[PTR:%.*]], align 16
+; CHECK-NEXT:    [[CAST_TO_FIXED:%.*]] = bitcast <vscale x 16 x i1>* [[PTR]] to <8 x i8>*
+; CHECK-NEXT:    [[A:%.*]] = load <8 x i8>, <8 x i8>* [[CAST_TO_FIXED]], align 16
+; CHECK-NEXT:    ret <8 x i8> [[A]]
+;
+  store <vscale x 16 x i1> zeroinitializer, <vscale x 16 x i1>* %ptr, align 16
+  %cast_to_fixed = bitcast <vscale x 16 x i1>* %ptr to <8 x i8>*
+  %a = load <8 x i8>, <8 x i8>* %cast_to_fixed, align 16
+  ret <8 x i8> %a
+}
Index: llvm/lib/Analysis/ConstantFolding.cpp
===================================================================
--- llvm/lib/Analysis/ConstantFolding.cpp
+++ llvm/lib/Analysis/ConstantFolding.cpp
@@ -352,9 +352,9 @@
                                          const DataLayout &DL) {
   do {
     Type *SrcTy = C->getType();
-    uint64_t DestSize = DL.getTypeSizeInBits(DestTy);
-    uint64_t SrcSize = DL.getTypeSizeInBits(SrcTy);
-    if (SrcSize < DestSize)
+    TypeSize DestSize = DL.getTypeSizeInBits(DestTy);
+    TypeSize SrcSize = DL.getTypeSizeInBits(SrcTy);
+    if (!TypeSize::isKnownGE(SrcSize, DestSize))
       return nullptr;
 
     // Catch the obvious splat cases (since all-zeros can coerce non-integral


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D112142.382995.patch
Type: text/x-patch
Size: 2268 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20211028/81f6ebb8/attachment.bin>