diff options
| -rw-r--r-- | llvm/lib/Transforms/Scalar/RewriteStatepointsForGC.cpp | 5 | ||||
| -rw-r--r-- | llvm/test/Transforms/RewriteStatepointsForGC/vector-bitcast.ll | 26 |
2 files changed, 31 insertions, 0 deletions
diff --git a/llvm/lib/Transforms/Scalar/RewriteStatepointsForGC.cpp b/llvm/lib/Transforms/Scalar/RewriteStatepointsForGC.cpp index 4b8ddb7cc24..1ca77cfec32 100644 --- a/llvm/lib/Transforms/Scalar/RewriteStatepointsForGC.cpp +++ b/llvm/lib/Transforms/Scalar/RewriteStatepointsForGC.cpp @@ -421,6 +421,11 @@ findBaseDefiningValueOfVector(Value *I) { if (auto *GEP = dyn_cast<GetElementPtrInst>(I)) return findBaseDefiningValue(GEP->getPointerOperand()); + // If the pointer comes through a bitcast of a vector of pointers to + // a vector of another type of pointer, then look through the bitcast + if (auto *BC = dyn_cast<BitCastInst>(I)) + return findBaseDefiningValue(BC->getOperand(0)); + // A PHI or Select is a base defining value. The outer findBasePointer // algorithm is responsible for constructing a base value for this BDV. assert((isa<SelectInst>(I) || isa<PHINode>(I)) && diff --git a/llvm/test/Transforms/RewriteStatepointsForGC/vector-bitcast.ll b/llvm/test/Transforms/RewriteStatepointsForGC/vector-bitcast.ll new file mode 100644 index 00000000000..981942a91ee --- /dev/null +++ b/llvm/test/Transforms/RewriteStatepointsForGC/vector-bitcast.ll @@ -0,0 +1,26 @@ +; RUN: opt -S -rewrite-statepoints-for-gc < %s | FileCheck %s +; +; A test to make sure that we can look through bitcasts of +; vector types when a base pointer is contained in a vector. + +target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128-ni:1" +target triple = "x86_64-unknown-linux-gnu" + +; Function Attrs: uwtable +define void @test() gc "statepoint-example" { +; CHECK-LABEL: @test +entry: +; CHECK-LABEL: entry +; CHECK: %bc = bitcast +; CHECK: %[[p1:[A-Za-z0-9_]+]] = extractelement +; CHECK: %[[p2:[A-Za-z0-9_]+]] = extractelement +; CHECK: llvm.experimental.gc.statepoint +; CHECK: %[[p2]].relocated = {{.+}} @llvm.experimental.gc.relocate +; CHECK: %[[p1]].relocated = {{.+}} @llvm.experimental.gc.relocate +; CHECK: load atomic + %bc = bitcast <8 x i8 addrspace(1)*> undef to <8 x i32 addrspace(1)*> + %ptr= extractelement <8 x i32 addrspace(1)*> %bc, i32 7 + %0 = call i8 addrspace(1)* undef() [ "deopt"() ] + %1 = load atomic i32, i32 addrspace(1)* %ptr unordered, align 4 + unreachable +} |
