fix test

Author: ellishg

llvm/test/CodeGen/AArch64/machine-outliner-loh.ll

@@ -9,47 +9,58 @@ declare void @bar(ptr %a);
 declare void @goo(ptr %a);
 
 ; CHECK-LABEL: _a0:
-define i32 @a0(i32 %a) {
+define void @a0(i32 %a) {
+
+  ; This becomes AdrpAdd when outlining is disabled, otherwise it is outlined
+  ; and there should be no LOH.
   %addr = getelementptr inbounds i32, ptr @A, i32 0
   %res = load i32, ptr %addr, align 4
   ; LOH:      [[L0:Lloh.+]]:
   ; LOH-NEXT:   adrp x19, _A@PAGE
   ; LOH-NEXT: [[L1:Lloh.+]]:
   ; LOH-NEXT:   add x19, x19, _A@PAGEOFF
+
   call void @foo()
   ; OUTLINE:      bl _OUTLINED_FUNCTION_0
   ; OUTLINE-NEXT: mov x0, x19
   ; OUTLINE-NEXT: bl _bar
   call void @bar(ptr %addr)
+
+  ; This becomes AdrpAddStr.
   %addr2 = getelementptr inbounds i32, ptr @B, i32 4
-  store i32 0, ptr %addr2, align 4
+  store i32 %res, ptr %addr2, align 4
   ; CHECK:      [[L2:Lloh.+]]:
   ; CHECK-NEXT:   adrp x8, _B@PAGE
   ; CHECK-NEXT: [[L3:Lloh.+]]:
   ; CHECK-NEXT:   add x8, x8, _B@PAGEOFF
-  ; CHECK-NEXT:   mov w0, w20
   ; CHECK-NEXT: [[L4:Lloh.+]]:
-  ; CHECK-NEXT:   str wzr, [x8, #16]
-  ret i32 %res
+  ; CHECK-NEXT:   str w20, [x8, #16]
+  ret void
+
   ; LOH-DAG:   .loh AdrpAdd [[L0]], [[L1]]
   ; CHECK-DAG: .loh AdrpAddStr [[L2]], [[L3]], [[L4]]
   ; CHECK:     .cfi_endproc
 }
 
 ; CHECK-LABEL: _a1:
 define i32 @a1(i32 %a) {
+
+  ; This becomes AdrpAdd when outlining is disabled, otherwise it is outlined
+  ; and there should be no LOH.
   %addr = getelementptr inbounds i32, ptr @A, i32 0
   %res = load i32, ptr %addr, align 4
   ; LOH:      [[L5:Lloh.+]]:
   ; LOH-NEXT:   adrp x19, _A@PAGE
   ; LOH-NEXT: [[L6:Lloh.+]]:
   ; LOH-NEXT:   add x19, x19, _A@PAGEOFF
+
   call void @foo()
   ; OUTLINE:      bl _OUTLINED_FUNCTION_0
   ; OUTLINE-NEXT: mov x0, x19
   ; OUTLINE-NEXT: bl _goo
   call void @goo(ptr %addr)
   ret i32 %res
+
   ; LOH:   .loh AdrpAdd [[L5]], [[L6]]
   ; CHECK: .cfi_endproc
 }