[llvm] 059cfe3 - [FuncAttrs] Add tests for willreturn callsite inference.

[Florian Hahn via llvm-commits]

Author: Florian Hahn
Date: 2021-02-18T11:41:42Z
New Revision: 059cfe30939db19ed042c80c8cba349f8a4d3c7f

URL: https://github.com/llvm/llvm-project/commit/059cfe30939db19ed042c80c8cba349f8a4d3c7f
DIFF: https://github.com/llvm/llvm-project/commit/059cfe30939db19ed042c80c8cba349f8a4d3c7f.diff

LOG: [FuncAttrs] Add tests for willreturn callsite inference.

Added: 
    llvm/test/Transforms/FunctionAttrs/willreturn-callsites.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/FunctionAttrs/willreturn-callsites.ll b/llvm/test/Transforms/FunctionAttrs/willreturn-callsites.ll
new file mode 100644
index 000000000000..6c9cef7c9b6e
--- /dev/null
+++ b/llvm/test/Transforms/FunctionAttrs/willreturn-callsites.ll
@@ -0,0 +1,64 @@
+; RUN: opt -function-attrs -S %s | FileCheck %s
+
+declare void @decl_readonly() readonly
+declare void @decl_readnone() readnone
+declare void @decl_argmemonly(i32*) argmemonly
+declare void @decl_unknown()
+
+define void @test_fn_mustprogress(i32* %ptr) mustprogress {
+; CHECK: Function Attrs: mustprogress
+; CHECK-LABEL: @test_fn_mustprogress(
+; CHECK-NOT:     call void @decl_readonly() #
+; CHECK-NOT:     call void @decl_readnone() #
+; CHECK-NOT:     call void @decl_unknown() #
+; CHECK-NOT:     call void @decl_argmemonly(i32* [[PTR:%.*]]) #
+; CHECK:         ret void
+;
+  call void @decl_readonly()
+  call void @decl_readnone()
+  call void @decl_unknown()
+  call void @decl_argmemonly(i32* %ptr)
+  ret void
+}
+
+define void @test_fn_willreturn(i32* %ptr) willreturn {
+; CHECK: Function Attrs: willreturn
+; CHECK-LABEL: @test_fn_willreturn(
+; CHECK-NOT:     call void @decl_readonly() #
+; CHECK-NOT :    call void @decl_readnone() #
+; CHECK-NOT:     call void @decl_unknown() #
+; CHECK-NOT:     call void @decl_argmemonly(i32* [[PTR:%.*]]) #
+; CHECK:         ret void
+;
+  call void @decl_readonly()
+  call void @decl_readnone()
+  call void @decl_unknown()
+  call void @decl_argmemonly(i32* %ptr)
+  ret void
+}
+
+define void @test_fn_mustprogress_readonly_calls(i32* %ptr) mustprogress {
+; CHECK: Function Attrs: readonly willreturn mustprogress
+; CHECK-LABEL: @test_fn_mustprogress_readonly_calls(
+; CHECK-NOT:     call void @decl_readonly() #
+; CHECK-NOT:     call void @decl_readnone() #
+; CHECK:         ret void
+;
+  call void @decl_readonly()
+  call void @decl_readnone()
+  ret void
+}
+
+define void @test_fn_mustprogress_readonly_calls_but_stores(i32* %ptr) mustprogress {
+; CHECK: Function Attrs: nofree mustprogress
+; CHECK-LABEL: @test_fn_mustprogress_readonly_calls_but_stores(
+; CHECK-NOT:     call void @decl_readonly() #
+; CHECK-NOT:     call void @decl_readnone() #
+; CHECK:         store i32 0, i32* [[PTR:%.*]], align 4
+; CHECK-NEXT:    ret void
+;
+  call void @decl_readonly()
+  call void @decl_readnone()
+  store i32 0, i32* %ptr
+  ret void
+}