[AIX] Enable dollar sign as PC in inlineasm

[jsji]

Summary

$ is used as PC for PowerPC inlineasm, ELF use it,
enable it for AIX XCOFF as well.

[nemanjai]

Is it possible to also add an MC test case for this (similar to test/MC/PowerPC/ppc64-operands.s)?

llvm/test/CodeGen/PowerPC/inline-asm-dollarpc.ll

4

Minor nit: IS isn't clear as a prefix, how about IAS for Integrated AS?

[jsji]

Address comments.

[nemanjai]

LGTM.

[amyk]

Also LGTM. Thanks Jinsong.

llvm/lib/Target/PowerPC/MCTargetDesc/PPCMCAsmInfo.cpp

PPCXCOFFMCAsmInfo::PPCXCOFFMCAsmInfo(bool Is64Bit, const Triple &T) {
   // A size of 8 is only supported by the assembler under 64-bit.
   Data64bitsDirective = Is64Bit ? "\t.vbyte\t8, " : nullptr;
 
   // Debug Information
   SupportsDebugInformation = true;
 
   // Set up DWARF directives
   MinInstAlignment = 4;
+
+  // Support $ as PC in inline asm
+  DollarIsPC = true;
 }

llvm/test/CodeGen/PowerPC/inline-asm-dollarpc.ll

+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
+; RUN: llc -mcpu=pwr7 -verify-machineinstrs \
+; RUN:     -mtriple=powerpc-unknown-aix < %s  | FileCheck %s --check-prefix=IAS32
+
+; RUN: llc -mcpu=pwr7 -verify-machineinstrs \
+; RUN:     -mtriple=powerpc64-unknown-aix < %s | FileCheck %s --check-prefix=IAS64
+
+; RUN: llc -mcpu=pwr7 -verify-machineinstrs -no-integrated-as \
+; RUN:     -mtriple=powerpc64-unknown-aix < %s | FileCheck %s --check-prefix=NOIAS
+
+define dso_local i32 @dollarpc() #0 {
+; IAS32-LABEL: dollarpc:
+; IAS32:       # %bb.0: # %entry
+; IAS32-NEXT:    li 3, 0
+; IAS32-NEXT:    stw 3, -4(1)
+; IAS32-NEXT:    #APP
+; IAS32-NEXT:    mfspr 5, 269
+; IAS32-NEXT:    mfspr 3, 268
+; IAS32-NEXT:    mfspr 4, 269
+; IAS32-NEXT:    cmpw 5, 4
+; IAS32-NEXT:  L..tmp0:
+; IAS32-NEXT:    bne 0, L..tmp0-16
+; IAS32-EMPTY:
+; IAS32-NEXT:    #NO_APP
+; IAS32-NEXT:    stw 3, -12(1)
+; IAS32-NEXT:    stw 4, -8(1)
+; IAS32-NEXT:    stw 5, -16(1)
+; IAS32-NEXT:    blr
+;
+; IAS64-LABEL: dollarpc:
+; IAS64:       # %bb.0: # %entry
+; IAS64-NEXT:    li 3, 0
+; IAS64-NEXT:    stw 3, -4(1)
+; IAS64-NEXT:    #APP
+; IAS64-NEXT:    mfspr 4, 269
+; IAS64-NEXT:    mfspr 3, 268
+; IAS64-NEXT:    mfspr 5, 269
+; IAS64-NEXT:    cmpw 4, 5
+; IAS64-NEXT:  L..tmp0:
+; IAS64-NEXT:    bne 0, L..tmp0-16
+; IAS64-EMPTY:
+; IAS64-NEXT:    #NO_APP
+; IAS64-NEXT:    stw 3, -12(1)
+; IAS64-NEXT:    rldimi 3, 5, 32, 0
+; IAS64-NEXT:    stw 5, -8(1)
+; IAS64-NEXT:    stw 4, -16(1)
+; IAS64-NEXT:    blr
+;
+; NOIAS-LABEL: dollarpc:
+; NOIAS:       # %bb.0: # %entry
+; NOIAS-NEXT:    li 3, 0
+; NOIAS-NEXT:    stw 3, -4(1)
+; NOIAS-NEXT:    #APP
+; NOIAS-NEXT:    mftbu 4
+; NOIAS-NEXT:  mftb 3
+; NOIAS-NEXT:  mftbu 5
+; NOIAS-NEXT:  cmpw 4,5
+; NOIAS-NEXT:  bne $-0x10
+; NOIAS-EMPTY:
+; NOIAS-NEXT:    #NO_APP
+; NOIAS-NEXT:    stw 3, -12(1)
+; NOIAS-NEXT:    rldimi 3, 5, 32, 0
+; NOIAS-NEXT:    stw 5, -8(1)
+; NOIAS-NEXT:    stw 4, -16(1)
+; NOIAS-NEXT:    blr
+entry:
+  %retval = alloca i32, align 4
+  %tbu = alloca i32, align 4
+  %tbl = alloca i32, align 4
+  %temp = alloca i32, align 4
+  store i32 0, i32* %retval, align 4
+  %0 = call { i32, i32, i32 } asm sideeffect "mftbu  $2\0Amftb   $0\0Amftbu  $1\0Acmpw   $2,$1\0Abne    $$-0x10\0A", "=r,=r,=r,~{cc}"()
+  %asmresult = extractvalue { i32, i32, i32 } %0, 0
+  %asmresult1 = extractvalue { i32, i32, i32 } %0, 1
+  %asmresult2 = extractvalue { i32, i32, i32 } %0, 2
+  store i32 %asmresult, i32* %tbl, align 4
+  store i32 %asmresult1, i32* %tbu, align 4
+  store i32 %asmresult2, i32* %temp, align 4
+  %1 = load i32, i32* %tbu, align 4
+  %conv = zext i32 %1 to i64
+  %shl = shl i64 %conv, 32
+  %2 = load i32, i32* %tbl, align 4
+  %conv3 = zext i32 %2 to i64
+  %or = or i64 %shl, %conv3
+  %conv4 = trunc i64 %or to i32
+  ret i32 %conv4
+}
+

llvm/test/MC/PowerPC/dollarpc.s

+# RUN: llvm-mc -triple powerpc64-unknown-unknown  --show-encoding %s | FileCheck -check-prefix=CHECK-BE %s
+# RUN: llvm-mc -triple powerpc64le-unknown-unknown  --show-encoding %s | FileCheck -check-prefix=CHECK-LE %s
+# RUN: llvm-mc -triple powerpc-aix-ibm-xcoff  --show-encoding %s 2>&1 | FileCheck -check-prefix=CHECK-BE %s
+
+# CHECK-BE:                                 # encoding: [0x42,0x00,A,0bAAAAAA00]
+# CHECK-LE:                                 # encoding: [0bAAAAAA00,A,0x00,0x42]
+            bdnz $-8
+