[NaryReassociate] Improve test CHECK

Add `CHECK` directives for the function calls.

Differential Revision: http://reviews.llvm.org/D12885

Patch by: Volkan Keles <vkeles@apple.com>

From: Mehdi Amini <mehdi.amini@apple.com>
llvm-svn: 247774

2 files changed, 6 insertions, 3 deletions

diff --git a/llvm/test/Transforms/NaryReassociate/nary-add.ll b/llvm/test/Transforms/NaryReassociate/nary-add.ll
index b3093ff6ecd..654ef2c4961 100644
--- a/llvm/test/Transforms/NaryReassociate/nary-add.ll
+++ b/llvm/test/Transforms/NaryReassociate/nary-add.ll
@@ -17,8 +17,9 @@ define void @left_reassociate(i32 %a, i32 %b, i32 %c) {
   call void @foo(i32 %1)
   %2 = add i32 %b, %c
   %3 = add i32 %a, %2
-; CHECK: add i32 [[BASE]], %b
+; CHECK: [[RESULT:%[a-zA-Z0-9]+]] = add i32 [[BASE]], %b
   call void @foo(i32 %3)
+; CHECK-NEXT: call void @foo(i32 [[RESULT]])
   ret void
 }
 
@@ -35,8 +36,9 @@ define void @right_reassociate(i32 %a, i32 %b, i32 %c) {
   call void @foo(i32 %1)
   %2 = add i32 %a, %b
   %3 = add i32 %2, %c
-; CHECK: add i32 [[BASE]], %b
+; CHECK: [[RESULT:%[a-zA-Z0-9]+]] = add i32 [[BASE]], %b
   call void @foo(i32 %3)
+; CHECK-NEXT: call void @foo(i32 [[RESULT]])
   ret void
 }
 
diff --git a/llvm/test/Transforms/NaryReassociate/nary-mul.ll b/llvm/test/Transforms/NaryReassociate/nary-mul.ll
index 3ec6615c2bd..467843c7a39 100644
--- a/llvm/test/Transforms/NaryReassociate/nary-mul.ll
+++ b/llvm/test/Transforms/NaryReassociate/nary-mul.ll
@@ -11,8 +11,9 @@ define void @bar(i32 %a, i32 %b, i32 %c) {
   call void @foo(i32 %1)
   %2 = mul i32 %a, %b
   %3 = mul i32 %2, %c
-; CHECK: mul i32 [[BASE]], %b
+; CHECK: [[RESULT:%[a-zA-Z0-9]+]] = mul i32 [[BASE]], %b
   call void @foo(i32 %3)
+; CHECK-NEXT: call void @foo(i32 [[RESULT]])
   ret void
 }