diff options
-rw-r--r-- | llvm/lib/Transforms/IPO/FunctionAttrs.cpp | 2 | ||||
-rw-r--r-- | llvm/test/Transforms/FunctionAttrs/readattrs.ll | 38 |
2 files changed, 39 insertions, 1 deletions
diff --git a/llvm/lib/Transforms/IPO/FunctionAttrs.cpp b/llvm/lib/Transforms/IPO/FunctionAttrs.cpp index edebc26d475..8a697c3e6af 100644 --- a/llvm/lib/Transforms/IPO/FunctionAttrs.cpp +++ b/llvm/lib/Transforms/IPO/FunctionAttrs.cpp @@ -154,7 +154,7 @@ static MemoryAccessKind checkFunctionMemoryAccess(Function &F, AAResults &AAR, for (CallSite::arg_iterator CI = CS.arg_begin(), CE = CS.arg_end(); CI != CE; ++CI) { Value *Arg = *CI; - if (!Arg->getType()->isPointerTy()) + if (!Arg->getType()->isPtrOrPtrVectorTy()) continue; AAMDNodes AAInfo; diff --git a/llvm/test/Transforms/FunctionAttrs/readattrs.ll b/llvm/test/Transforms/FunctionAttrs/readattrs.ll index 7f22e6f2a2c..aabdfe8d200 100644 --- a/llvm/test/Transforms/FunctionAttrs/readattrs.ll +++ b/llvm/test/Transforms/FunctionAttrs/readattrs.ll @@ -65,3 +65,41 @@ entry: store i32 10, i32* %call, align 4 ret void } + +; CHECK: declare void @llvm.masked.scatter +declare void @llvm.masked.scatter.v4i32(<4 x i32>%val, <4 x i32*>, i32, <4 x i1>) + +; CHECK-NOT: readnone +; CHECK-NOT: readonly +; CHECK: define void @test9 +define void @test9(<4 x i32*> %ptrs, <4 x i32>%val) { + call void @llvm.masked.scatter.v4i32(<4 x i32>%val, <4 x i32*> %ptrs, i32 4, <4 x i1><i1 true, i1 false, i1 true, i1 false>) + ret void +} + +; CHECK: declare <4 x i32> @llvm.masked.gather +declare <4 x i32> @llvm.masked.gather.v4i32(<4 x i32*>, i32, <4 x i1>, <4 x i32>) +; CHECK: readonly +; CHECK: define <4 x i32> @test10 +define <4 x i32> @test10(<4 x i32*> %ptrs) { + %res = call <4 x i32> @llvm.masked.gather.v4i32(<4 x i32*> %ptrs, i32 4, <4 x i1><i1 true, i1 false, i1 true, i1 false>, <4 x i32>undef) + ret <4 x i32> %res +} + +; CHECK: declare <4 x i32> @test11_1 +declare <4 x i32> @test11_1(<4 x i32*>) argmemonly nounwind readonly +; CHECK: readonly +; CHECK-NOT: readnone +; CHECK: define <4 x i32> @test11_2 +define <4 x i32> @test11_2(<4 x i32*> %ptrs) { + %res = call <4 x i32> @test11_1(<4 x i32*> %ptrs) + ret <4 x i32> %res +} + +declare <4 x i32> @test12_1(<4 x i32*>) argmemonly nounwind +; CHECK-NOT: readnone +; CHECK: define <4 x i32> @test12_2 +define <4 x i32> @test12_2(<4 x i32*> %ptrs) { + %res = call <4 x i32> @test12_1(<4 x i32*> %ptrs) + ret <4 x i32> %res +} |