commit dfddd3c4f8e104fde218776b2da95f3601e103ad
author Florian Hahn <flo@fhahn.com> Thu Feb 18 11:13:05 2021 +0000
committer Copybara-Service <copybara-worker@google.com> Fri Feb 19 03:28:33 2021 -0800
tree 2b113075320331b59862c03fd7cb46ca3fab05dd
parent 4d0e976c7fa627bdf6bef649327f44ae8a5c2533

[FuncAttrs] Add tests for willreturn callsite inference.

GitOrigin-RevId: 059cfe30939db19ed042c80c8cba349f8a4d3c7f

test/Transforms/FunctionAttrs/willreturn-callsites.ll

diff --git a/test/Transforms/FunctionAttrs/willreturn-callsites.ll b/test/Transforms/FunctionAttrs/willreturn-callsites.ll
new file mode 100644
index 0000000..6c9cef7
--- /dev/null
+++ b/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
+}