[AArch64] Update v8.5a MTE instructions

[javed.absar]

Summary

The latest MTE specification adds register Xt to the STG instruction family:

STG [Xn, #offset] -> STG Xt, [Xn, #offset]

The tag written to memory is taken from Xt rather than Xn.

The LDG instruction also was changed to read return address from Xt:

LDG Xt, [Xn, #offset].

This patch includes those changes and tests.

[DavidSpickett]

LGTM but add the specification link in the commit msg (https://developer.arm.com/docs/ddi0596/c) and reference LDG/STG in the title.

"Update v8.5a MTE LDG and STG instructions." would be fine.

lib/Target/AArch64/AArch64InstrFormats.td

class MemTagVector<bit Load, string asm_insn, string asm_opnds,
 
   let Inst{20-12} = 0b000000000;
   let Inst{4-0}   = Rt;
 
   let mayLoad = Load;
 }
 
 class MemTagLoad<string asm_insn, string asm_opnds>
-    : BaseMemTag<0b01, 0b00, asm_insn, asm_opnds, "", (outs GPR64:$Rt),
-                 (ins GPR64sp:$Rn, simm9s16:$offset)> {
+    : BaseMemTag<0b01, 0b00, asm_insn, asm_opnds, "$Rt = $wback",
+                 (outs GPR64:$wback),
+                 (ins GPR64:$Rt, GPR64sp:$Rn, simm9s16:$offset)> {
   bits<5> Rt;
   bits<9> offset;
 
   let Inst{20-12} = offset;
   let Inst{4-0}   = Rt;
 
   let mayLoad = 1;
 }
 
 class BaseMemTagStore<bits<2> opc1, bits<2> opc2, string asm_insn,
                      string asm_opnds, string cstr, dag oops, dag iops>
     : BaseMemTag<opc1, opc2, asm_insn, asm_opnds, cstr, oops, iops> {
   bits<5> Rt;
   bits<9> offset;
 
   let Inst{20-12} = offset;
-  let Inst{4-0}   = 0b11111;
-  let Unpredictable{4-0} = 0b11111;
+  let Inst{4-0}   = Rt;
 
   let mayStore = 1;
 }
 
 multiclass MemTagStore<bits<2> opc1, string insn> {
   def Offset :
-    BaseMemTagStore<opc1, 0b10, insn, "\t[$Rn, $offset]", "",
-                    (outs), (ins GPR64sp:$Rn, simm9s16:$offset)>;
+    BaseMemTagStore<opc1, 0b10, insn, "\t$Rt, [$Rn, $offset]", "",
+                    (outs), (ins GPR64sp:$Rt, GPR64sp:$Rn, simm9s16:$offset)>;
   def PreIndex :
-    BaseMemTagStore<opc1, 0b11, insn, "\t[$Rn, $offset]!",
+    BaseMemTagStore<opc1, 0b11, insn, "\t$Rt, [$Rn, $offset]!",
                     "$Rn = $wback,@earlyclobber $wback",
                     (outs GPR64sp:$wback),
-                    (ins GPR64sp:$Rn, simm9s16:$offset)>;
+                    (ins GPR64sp:$Rt, GPR64sp:$Rn, simm9s16:$offset)>;
   def PostIndex :
-    BaseMemTagStore<opc1, 0b01, insn, "\t[$Rn], $offset",
+    BaseMemTagStore<opc1, 0b01, insn, "\t$Rt, [$Rn], $offset",
                     "$Rn = $wback,@earlyclobber $wback",
                     (outs GPR64sp:$wback),
-                    (ins GPR64sp:$Rn, simm9s16:$offset)>;
+                    (ins GPR64sp:$Rt, GPR64sp:$Rn, simm9s16:$offset)>;
 
-  def : InstAlias<insn # "\t[$Rn]",
-                  (!cast<Instruction>(NAME # "Offset") GPR64sp:$Rn, 0)>;
+  def : InstAlias<insn # "\t$Rt, [$Rn]",
+                  (!cast<Instruction>(NAME # "Offset") GPR64sp:$Rt, GPR64sp:$Rn, 0)>;
 }
 
 //---
 // Exception generation
 //---
 
 let mayLoad = 0, mayStore = 0, hasSideEffects = 1 in
 class ExceptionGeneration<bits<3> op1, bits<2> ll, string asm>

test/MC/AArch64/armv8.5a-mte.s

 // NOMTE-NEXT: gmi x0, x1, x2
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: gmi x3, sp, x4
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: gmi xzr, x0, x30
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: gmi x30, x0, xzr
 
-stg [x1]
-stg [x1, #-4096]
-stg [x2, #4080]
-stg [sp, #16]
-
-// CHECK: stg [x1]              // encoding: [0x3f,0x08,0x20,0xd9]
-// CHECK: stg [x1, #-4096]      // encoding: [0x3f,0x08,0x30,0xd9]
-// CHECK: stg [x2, #4080]       // encoding: [0x5f,0xf8,0x2f,0xd9]
-// CHECK: stg [sp, #16]         // encoding: [0xff,0x1b,0x20,0xd9]
+stg x0,  [x1]
+stg x1,  [x1, #-4096]
+stg x2,  [x2, #4080]
+stg x3,  [sp, #16]
+stg sp,  [sp, #16]
+
+// CHECK: stg x0,  [x1]              // encoding: [0x20,0x08,0x20,0xd9]
+// CHECK: stg x1,  [x1, #-4096]      // encoding: [0x21,0x08,0x30,0xd9]
+// CHECK: stg x2,  [x2, #4080]       // encoding: [0x42,0xf8,0x2f,0xd9]
+// CHECK: stg x3,  [sp, #16]         // encoding: [0xe3,0x1b,0x20,0xd9]
+// CHECK: stg sp,  [sp, #16]         // encoding: [0xff,0x1b,0x20,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stg
 
-stzg [x1]
-stzg [x1, #-4096]
-stzg [x2, #4080]
-stzg [sp, #16]
-
-// CHECK: stzg [x1]             // encoding: [0x3f,0x08,0x60,0xd9]
-// CHECK: stzg [x1, #-4096]     // encoding: [0x3f,0x08,0x70,0xd9]
-// CHECK: stzg [x2, #4080]      // encoding: [0x5f,0xf8,0x6f,0xd9]
-// CHECK: stzg [sp, #16]        // encoding: [0xff,0x1b,0x60,0xd9]
+stzg x0,  [x1]
+stzg x1,  [x1, #-4096]
+stzg x2,  [x2, #4080]
+stzg x3,  [sp, #16]
+stzg sp,  [sp, #16]
+
+// CHECK: stzg x0,  [x1]             // encoding: [0x20,0x08,0x60,0xd9]
+// CHECK: stzg x1,  [x1, #-4096]     // encoding: [0x21,0x08,0x70,0xd9]
+// CHECK: stzg x2,  [x2, #4080]      // encoding: [0x42,0xf8,0x6f,0xd9]
+// CHECK: stzg x3,  [sp, #16]        // encoding: [0xe3,0x1b,0x60,0xd9]
+// CHECK: stzg sp,  [sp, #16]        // encoding: [0xff,0x1b,0x60,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stzg
 
-stg [x1, #-4096]!
-stg [x2, #4080]!
-stg [sp, #16]!
-
-// CHECK: stg [x1, #-4096]!      // encoding: [0x3f,0x0c,0x30,0xd9]
-// CHECK: stg [x2, #4080]!       // encoding: [0x5f,0xfc,0x2f,0xd9]
-// CHECK: stg [sp, #16]!         // encoding: [0xff,0x1f,0x20,0xd9]
+stg x0,  [x1, #-4096]!
+stg x1,  [x2, #4080]!
+stg x2,  [sp, #16]!
+stg sp,  [sp, #16]!
+
+// CHECK: stg x0,  [x1, #-4096]!      // encoding: [0x20,0x0c,0x30,0xd9]
+// CHECK: stg x1,  [x2, #4080]!       // encoding: [0x41,0xfc,0x2f,0xd9]
+// CHECK: stg x2,  [sp, #16]!         // encoding: [0xe2,0x1f,0x20,0xd9]
+// CHECK: stg sp,  [sp, #16]!         // encoding: [0xff,0x1f,0x20,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stg
 
-stzg [x1, #-4096]!
-stzg [x2, #4080]!
-stzg [sp, #16]!
-
-// CHECK: stzg [x1, #-4096]!     // encoding: [0x3f,0x0c,0x70,0xd9]
-// CHECK: stzg [x2, #4080]!      // encoding: [0x5f,0xfc,0x6f,0xd9]
-// CHECK: stzg [sp, #16]!        // encoding: [0xff,0x1f,0x60,0xd9]
+stzg x0,  [x1, #-4096]!
+stzg x1,  [x2, #4080]!
+stzg x2,  [sp, #16]!
+stzg sp,  [sp, #16]!
+
+// CHECK: stzg x0,  [x1, #-4096]!     // encoding: [0x20,0x0c,0x70,0xd9]
+// CHECK: stzg x1,  [x2, #4080]!      // encoding: [0x41,0xfc,0x6f,0xd9]
+// CHECK: stzg x2,  [sp, #16]!        // encoding: [0xe2,0x1f,0x60,0xd9]
+// CHECK: stzg sp,  [sp, #16]!        // encoding: [0xff,0x1f,0x60,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stzg
 
-stg [x1], #-4096
-stg [x2], #4080
-stg [sp], #16
-
-// CHECK: stg [x1], #-4096       // encoding: [0x3f,0x04,0x30,0xd9]
-// CHECK: stg [x2], #4080        // encoding: [0x5f,0xf4,0x2f,0xd9]
-// CHECK: stg [sp], #16          // encoding: [0xff,0x17,0x20,0xd9]
+stg x0,  [x1], #-4096
+stg x1,  [x2], #4080
+stg x2,  [sp], #16
+stg sp,  [sp], #16
+
+// CHECK: stg x0,  [x1], #-4096       // encoding: [0x20,0x04,0x30,0xd9]
+// CHECK: stg x1,  [x2], #4080        // encoding: [0x41,0xf4,0x2f,0xd9]
+// CHECK: stg x2,  [sp], #16          // encoding: [0xe2,0x17,0x20,0xd9]
+// CHECK: stg sp,  [sp], #16          // encoding: [0xff,0x17,0x20,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stg
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stg
 
-stzg [x1], #-4096
-stzg [x2], #4080
-stzg [sp], #16
-
-// CHECK: stzg [x1], #-4096      // encoding: [0x3f,0x04,0x70,0xd9]
-// CHECK: stzg [x2], #4080       // encoding: [0x5f,0xf4,0x6f,0xd9]
-// CHECK: stzg [sp], #16         // encoding: [0xff,0x17,0x60,0xd9]
+stzg x0,  [x1], #-4096
+stzg x1,  [x2], #4080
+stzg x2,  [sp], #16
+stzg sp,  [sp], #16
+
+// CHECK: stzg x0,  [x1], #-4096      // encoding: [0x20,0x04,0x70,0xd9]
+// CHECK: stzg x1,  [x2], #4080       // encoding: [0x41,0xf4,0x6f,0xd9]
+// CHECK: stzg x2,  [sp], #16         // encoding: [0xe2,0x17,0x60,0xd9]
+// CHECK: stzg sp,  [sp], #16         // encoding: [0xff,0x17,0x60,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stzg
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stzg
 
-st2g [x1]
-st2g [x1, #-4096]
-st2g [x2, #4080]
-st2g [sp, #16]
-
-// CHECK: st2g [x1]              // encoding: [0x3f,0x08,0xa0,0xd9]
-// CHECK: st2g [x1, #-4096]      // encoding: [0x3f,0x08,0xb0,0xd9]
-// CHECK: st2g [x2, #4080]       // encoding: [0x5f,0xf8,0xaf,0xd9]
-// CHECK: st2g [sp, #16]         // encoding: [0xff,0x1b,0xa0,0xd9]
+st2g x0,  [x1]
+st2g x1,  [x1, #-4096]
+st2g x2,  [x2, #4080]
+st2g x3,  [sp, #16]
+st2g sp,  [sp, #16]
+
+// CHECK: st2g x0,  [x1]              // encoding: [0x20,0x08,0xa0,0xd9]
+// CHECK: st2g x1,  [x1, #-4096]      // encoding: [0x21,0x08,0xb0,0xd9]
+// CHECK: st2g x2,  [x2, #4080]       // encoding: [0x42,0xf8,0xaf,0xd9]
+// CHECK: st2g x3,  [sp, #16]         // encoding: [0xe3,0x1b,0xa0,0xd9]
+// CHECK: st2g sp,  [sp, #16]         // encoding: [0xff,0x1b,0xa0,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: st2g
 
-stz2g [x1]
-stz2g [x1, #-4096]
-stz2g [x2, #4080]
-stz2g [sp, #16]
-
-// CHECK: stz2g [x1]             // encoding: [0x3f,0x08,0xe0,0xd9]
-// CHECK: stz2g [x1, #-4096]     // encoding: [0x3f,0x08,0xf0,0xd9]
-// CHECK: stz2g [x2, #4080]      // encoding: [0x5f,0xf8,0xef,0xd9]
-// CHECK: stz2g [sp, #16]        // encoding: [0xff,0x1b,0xe0,0xd9]
+stz2g x0,  [x1]
+stz2g x1,  [x1, #-4096]
+stz2g x2,  [x2, #4080]
+stz2g x3,  [sp, #16]
+stz2g sp,  [sp, #16]
+
+// CHECK: stz2g x0,  [x1]             // encoding: [0x20,0x08,0xe0,0xd9]
+// CHECK: stz2g x1,  [x1, #-4096]     // encoding: [0x21,0x08,0xf0,0xd9]
+// CHECK: stz2g x2,  [x2, #4080]      // encoding: [0x42,0xf8,0xef,0xd9]
+// CHECK: stz2g x3,  [sp, #16]        // encoding: [0xe3,0x1b,0xe0,0xd9]
+// CHECK: stz2g sp,  [sp, #16]        // encoding: [0xff,0x1b,0xe0,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stz2g
 
-st2g [x1, #-4096]!
-st2g [x2, #4080]!
-st2g [sp, #16]!
-
-// CHECK: st2g [x1, #-4096]!     // encoding: [0x3f,0x0c,0xb0,0xd9]
-// CHECK: st2g [x2, #4080]!      // encoding: [0x5f,0xfc,0xaf,0xd9]
-// CHECK: st2g [sp, #16]!        // encoding: [0xff,0x1f,0xa0,0xd9]
+st2g x0,  [x1, #-4096]!
+st2g x1,  [x2, #4080]!
+st2g x2,  [sp, #16]!
+st2g sp,  [sp, #16]!
+
+// CHECK: st2g x0,  [x1, #-4096]!     // encoding: [0x20,0x0c,0xb0,0xd9]
+// CHECK: st2g x1,  [x2, #4080]!      // encoding: [0x41,0xfc,0xaf,0xd9]
+// CHECK: st2g x2,  [sp, #16]!        // encoding: [0xe2,0x1f,0xa0,0xd9]
+// CHECK: st2g sp,  [sp, #16]!        // encoding: [0xff,0x1f,0xa0,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: st2g
 
-stz2g [x1, #-4096]!
-stz2g [x2, #4080]!
-stz2g [sp, #16]!
-
-// CHECK: stz2g [x1, #-4096]!    // encoding: [0x3f,0x0c,0xf0,0xd9]
-// CHECK: stz2g [x2, #4080]!     // encoding: [0x5f,0xfc,0xef,0xd9]
-// CHECK: stz2g [sp, #16]!       // encoding: [0xff,0x1f,0xe0,0xd9]
+stz2g x0,  [x1, #-4096]!
+stz2g x1,  [x2, #4080]!
+stz2g x2,  [sp, #16]!
+stz2g sp,  [sp, #16]!
+
+// CHECK: stz2g x0,  [x1, #-4096]!    // encoding: [0x20,0x0c,0xf0,0xd9]
+// CHECK: stz2g x1,  [x2, #4080]!     // encoding: [0x41,0xfc,0xef,0xd9]
+// CHECK: stz2g x2,  [sp, #16]!       // encoding: [0xe2,0x1f,0xe0,0xd9]
+// CHECK: stz2g sp,  [sp, #16]!       // encoding: [0xff,0x1f,0xe0,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stz2g
 
-st2g [x1], #-4096
-st2g [x2], #4080
-st2g [sp], #16
-
-// CHECK: st2g [x1], #-4096      // encoding: [0x3f,0x04,0xb0,0xd9]
-// CHECK: st2g [x2], #4080       // encoding: [0x5f,0xf4,0xaf,0xd9]
-// CHECK: st2g [sp], #16         // encoding: [0xff,0x17,0xa0,0xd9]
+st2g x0,  [x1], #-4096
+st2g x1,  [x2], #4080
+st2g x2,  [sp], #16
+st2g sp,  [sp], #16
+
+// CHECK: st2g x0,  [x1], #-4096      // encoding: [0x20,0x04,0xb0,0xd9]
+// CHECK: st2g x1,  [x2], #4080       // encoding: [0x41,0xf4,0xaf,0xd9]
+// CHECK: st2g x2,  [sp], #16         // encoding: [0xe2,0x17,0xa0,0xd9]
+// CHECK: st2g sp,  [sp], #16         // encoding: [0xff,0x17,0xa0,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: st2g
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: st2g
 
-stz2g [x1], #-4096
-stz2g [x2], #4080
-stz2g [sp], #16
-
-// CHECK: stz2g [x1], #-4096     // encoding: [0x3f,0x04,0xf0,0xd9]
-// CHECK: stz2g [x2], #4080      // encoding: [0x5f,0xf4,0xef,0xd9]
-// CHECK: stz2g [sp], #16        // encoding: [0xff,0x17,0xe0,0xd9]
+stz2g x0,  [x1], #-4096
+stz2g x1,  [x2], #4080
+stz2g x2,  [sp], #16
+stz2g sp,  [sp], #16
+
+// CHECK: stz2g x0,  [x1], #-4096     // encoding: [0x20,0x04,0xf0,0xd9]
+// CHECK: stz2g x1,  [x2], #4080      // encoding: [0x41,0xf4,0xef,0xd9]
+// CHECK: stz2g x2,  [sp], #16        // encoding: [0xe2,0x17,0xe0,0xd9]
+// CHECK: stz2g sp,  [sp], #16        // encoding: [0xff,0x17,0xe0,0xd9]
 
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
 // NOMTE:      instruction requires: mte
 // NOMTE-NEXT: stz2g
+// NOMTE:      instruction requires: mte
+// NOMTE-NEXT: stz2g
 
 stgp x0, x1, [x2]
 stgp x0, x1, [x2, #-1024]
 stgp x0, x1, [x2, #1008]
 stgp x0, x1, [sp, #16]
 stgp xzr, x1, [x2, #16]
 stgp x0, xzr, [x2, #16]
 

test/MC/Disassembler/AArch64/armv8.5a-mte.txt

 # NOMTE-NEXT: [0xe0,0x03,0xdf,0x9a]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xe0,0x03,0xdf,0xba]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0x1f,0x00,0xc1,0xba]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x03,0xdf,0xba]
 
-[0x3f,0x08,0x30,0xd9]
-[0x5f,0xf8,0x2f,0xd9]
-[0xff,0x1b,0x20,0xd9]
+[0x20,0x08,0x30,0xd9]
+[0x41,0xf8,0x2f,0xd9]
+[0xe2,0x1b,0x20,0xd9]
+[0x23,0x08,0x20,0xd9]
 [0x3f,0x08,0x20,0xd9]
 
-# CHECK: stg [x1, #-4096]
-# CHECK: stg [x2, #4080]
-# CHECK: stg [sp, #16]
-# CHECK: stg [x1]
+# CHECK: stg x0,  [x1, #-4096]
+# CHECK: stg x1,  [x2, #4080]
+# CHECK: stg x2,  [sp, #16]
+# CHECK: stg x3,  [x1]
+# CHECK: stg sp,  [x1]
 
+
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x20,0x08,0x30,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x08,0x30,0xd9]
+# NOMTE-NEXT: [0x41,0xf8,0x2f,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xf8,0x2f,0xd9]
+# NOMTE-NEXT: [0xe2,0x1b,0x20,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0xff,0x1b,0x20,0xd9]
+# NOMTE-NEXT: [0x23,0x08,0x20,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0x3f,0x08,0x20,0xd9]
 
-[0x3f,0x08,0x70,0xd9]
-[0x5f,0xf8,0x6f,0xd9]
-[0xff,0x1b,0x60,0xd9]
+[0x20,0x08,0x70,0xd9]
+[0x41,0xf8,0x6f,0xd9]
+[0xe2,0x1b,0x60,0xd9]
+[0x23,0x08,0x60,0xd9]
 [0x3f,0x08,0x60,0xd9]
 
-# CHECK: stzg [x1, #-4096]
-# CHECK: stzg [x2, #4080]
-# CHECK: stzg [sp, #16]
-# CHECK: stzg [x1]
+# CHECK: stzg x0,  [x1, #-4096]
+# CHECK: stzg x1,  [x2, #4080]
+# CHECK: stzg x2,  [sp, #16]
+# CHECK: stzg x3,  [x1]
+# CHECK: stzg sp,  [x1]
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x08,0x70,0xd9]
+# NOMTE-NEXT: [0x20,0x08,0x70,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xf8,0x6f,0xd9]
+# NOMTE-NEXT: [0x41,0xf8,0x6f,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0xff,0x1b,0x60,0xd9]
+# NOMTE-NEXT: [0xe2,0x1b,0x60,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x23,0x08,0x60,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0x3f,0x08,0x60,0xd9]
 
-[0x3f,0x0c,0x30,0xd9]
-[0x5f,0xfc,0x2f,0xd9]
+[0x20,0x0c,0x30,0xd9]
+[0x41,0xfc,0x2f,0xd9]
+[0xe2,0x1f,0x20,0xd9]
 [0xff,0x1f,0x20,0xd9]
 
-# CHECK: stg [x1, #-4096]!
-# CHECK: stg [x2, #4080]!
-# CHECK: stg [sp, #16]!
+# CHECK: stg x0,  [x1, #-4096]!
+# CHECK: stg x1,  [x2, #4080]!
+# CHECK: stg x2,  [sp, #16]!
+# CHECK: stg sp,  [sp, #16]!
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x0c,0x30,0xd9]
+# NOMTE-NEXT: [0x20,0x0c,0x30,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x41,0xfc,0x2f,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xfc,0x2f,0xd9]
+# NOMTE-NEXT: [0xe2,0x1f,0x20,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x1f,0x20,0xd9]
 
-[0x3f,0x0c,0x70,0xd9]
-[0x5f,0xfc,0x6f,0xd9]
+[0x20,0x0c,0x70,0xd9]
+[0x41,0xfc,0x6f,0xd9]
+[0xe2,0x1f,0x60,0xd9]
 [0xff,0x1f,0x60,0xd9]
 
-# CHECK: stzg [x1, #-4096]!
-# CHECK: stzg [x2, #4080]!
-# CHECK: stzg [sp, #16]!
+# CHECK: stzg x0,  [x1, #-4096]!
+# CHECK: stzg x1,  [x2, #4080]!
+# CHECK: stzg x2,  [sp, #16]!
+# CHECK: stzg sp,  [sp, #16]!
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x0c,0x70,0xd9]
+# NOMTE-NEXT: [0x20,0x0c,0x70,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xfc,0x6f,0xd9]
+# NOMTE-NEXT: [0x41,0xfc,0x6f,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0xe2,0x1f,0x60,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x1f,0x60,0xd9]
 
-[0x3f,0x04,0x30,0xd9]
-[0x5f,0xf4,0x2f,0xd9]
+[0x20,0x04,0x30,0xd9]
+[0x41,0xf4,0x2f,0xd9]
+[0xe2,0x17,0x20,0xd9]
 [0xff,0x17,0x20,0xd9]
 
-# CHECK: stg [x1], #-4096
-# CHECK: stg [x2], #4080
-# CHECK: stg [sp], #16
+# CHECK: stg x0,  [x1], #-4096
+# CHECK: stg x1,  [x2], #4080
+# CHECK: stg x2,  [sp], #16
+# CHECK: stg sp,  [sp], #16
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x04,0x30,0xd9]
+# NOMTE-NEXT: [0x20,0x04,0x30,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x41,0xf4,0x2f,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xf4,0x2f,0xd9]
+# NOMTE-NEXT: [0xe2,0x17,0x20,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x17,0x20,0xd9]
 
-[0x3f,0x04,0x70,0xd9]
-[0x5f,0xf4,0x6f,0xd9]
+[0x20,0x04,0x70,0xd9]
+[0x41,0xf4,0x6f,0xd9]
+[0xe2,0x17,0x60,0xd9]
 [0xff,0x17,0x60,0xd9]
 
-# CHECK: stzg [x1], #-4096
-# CHECK: stzg [x2], #4080
-# CHECK: stzg [sp], #16
+# CHECK: stzg x0,  [x1], #-4096
+# CHECK: stzg x1,  [x2], #4080
+# CHECK: stzg x2,  [sp], #16
+# CHECK: stzg sp,  [sp], #16
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x04,0x70,0xd9]
+# NOMTE-NEXT: [0x20,0x04,0x70,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x41,0xf4,0x6f,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xf4,0x6f,0xd9]
+# NOMTE-NEXT: [0xe2,0x17,0x60,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x17,0x60,0xd9]
 
-[0x3f,0x08,0xb0,0xd9]
-[0x5f,0xf8,0xaf,0xd9]
-[0xff,0x1b,0xa0,0xd9]
+[0x20,0x08,0xb0,0xd9]
+[0x41,0xf8,0xaf,0xd9]
+[0xe2,0x1b,0xa0,0xd9]
+[0x23,0x08,0xa0,0xd9]
 [0x3f,0x08,0xa0,0xd9]
 
-# CHECK: st2g [x1, #-4096]
-# CHECK: st2g [x2, #4080]
-# CHECK: st2g [sp, #16]
-# CHECK: st2g [x1]
+# CHECK: st2g x0,  [x1, #-4096]
+# CHECK: st2g x1,  [x2, #4080]
+# CHECK: st2g x2,  [sp, #16]
+# CHECK: st2g x3,  [x1]
+# CHECK: st2g sp,  [x1]
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x08,0xb0,0xd9]
+# NOMTE-NEXT: [0x20,0x08,0xb0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xf8,0xaf,0xd9]
+# NOMTE-NEXT: [0x41,0xf8,0xaf,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0xff,0x1b,0xa0,0xd9]
+# NOMTE-NEXT: [0xe2,0x1b,0xa0,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x23,0x08,0xa0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0x3f,0x08,0xa0,0xd9]
 
-[0x3f,0x08,0xf0,0xd9]
-[0x5f,0xf8,0xef,0xd9]
-[0xff,0x1b,0xe0,0xd9]
+[0x20,0x08,0xf0,0xd9]
+[0x41,0xf8,0xef,0xd9]
+[0xe2,0x1b,0xe0,0xd9]
+[0x23,0x08,0xe0,0xd9]
 [0x3f,0x08,0xe0,0xd9]
 
-# CHECK: stz2g [x1, #-4096]
-# CHECK: stz2g [x2, #4080]
-# CHECK: stz2g [sp, #16]
-# CHECK: stz2g [x1]
+# CHECK: stz2g x0,  [x1, #-4096]
+# CHECK: stz2g x1,  [x2, #4080]
+# CHECK: stz2g x2,  [sp, #16]
+# CHECK: stz2g x3,  [x1]
+# CHECK: stz2g sp,  [x1]
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x08,0xf0,0xd9]
+# NOMTE-NEXT: [0x20,0x08,0xf0,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x41,0xf8,0xef,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xf8,0xef,0xd9]
+# NOMTE-NEXT: [0xe2,0x1b,0xe0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0xff,0x1b,0xe0,0xd9]
+# NOMTE-NEXT: [0x23,0x08,0xe0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0x3f,0x08,0xe0,0xd9]
 
-[0x3f,0x0c,0xb0,0xd9]
-[0x5f,0xfc,0xaf,0xd9]
+[0x20,0x0c,0xb0,0xd9]
+[0x41,0xfc,0xaf,0xd9]
+[0xe2,0x1f,0xa0,0xd9]
 [0xff,0x1f,0xa0,0xd9]
 
-# CHECK: st2g [x1, #-4096]!
-# CHECK: st2g [x2, #4080]!
-# CHECK: st2g [sp, #16]!
+# CHECK: st2g x0,  [x1, #-4096]!
+# CHECK: st2g x1,  [x2, #4080]!
+# CHECK: st2g x2,  [sp, #16]!
+# CHECK: st2g sp,  [sp, #16]!
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x0c,0xb0,0xd9]
+# NOMTE-NEXT: [0x20,0x0c,0xb0,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x41,0xfc,0xaf,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xfc,0xaf,0xd9]
+# NOMTE-NEXT: [0xe2,0x1f,0xa0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x1f,0xa0,0xd9]
 
-[0x3f,0x0c,0xf0,0xd9]
-[0x5f,0xfc,0xef,0xd9]
+[0x20,0x0c,0xf0,0xd9]
+[0x41,0xfc,0xef,0xd9]
+[0xe2,0x1f,0xe0,0xd9]
 [0xff,0x1f,0xe0,0xd9]
 
-# CHECK: stz2g [x1, #-4096]!
-# CHECK: stz2g [x2, #4080]!
-# CHECK: stz2g [sp, #16]!
+# CHECK: stz2g x0,  [x1, #-4096]!
+# CHECK: stz2g x1,  [x2, #4080]!
+# CHECK: stz2g x2,  [sp, #16]!
+# CHECK: stz2g sp,  [sp, #16]!
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x0c,0xf0,0xd9]
+# NOMTE-NEXT: [0x20,0x0c,0xf0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xfc,0xef,0xd9]
+# NOMTE-NEXT: [0x41,0xfc,0xef,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0xe2,0x1f,0xe0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x1f,0xe0,0xd9]
 
-[0x3f,0x04,0xb0,0xd9]
-[0x5f,0xf4,0xaf,0xd9]
+[0x20,0x04,0xb0,0xd9]
+[0x41,0xf4,0xaf,0xd9]
+[0xe2,0x17,0xa0,0xd9]
 [0xff,0x17,0xa0,0xd9]
 
-# CHECK: st2g [x1], #-4096
-# CHECK: st2g [x2], #4080
-# CHECK: st2g [sp], #16
+# CHECK: st2g x0,  [x1], #-4096
+# CHECK: st2g x1,  [x2], #4080
+# CHECK: st2g x2,  [sp], #16
+# CHECK: st2g sp,  [sp], #16
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x04,0xb0,0xd9]
+# NOMTE-NEXT: [0x20,0x04,0xb0,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x41,0xf4,0xaf,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xf4,0xaf,0xd9]
+# NOMTE-NEXT: [0xe2,0x17,0xa0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x17,0xa0,0xd9]
 
-[0x3f,0x04,0xf0,0xd9]
-[0x5f,0xf4,0xef,0xd9]
+[0x20,0x04,0xf0,0xd9]
+[0x41,0xf4,0xef,0xd9]
+[0xe2,0x17,0xe0,0xd9]
 [0xff,0x17,0xe0,0xd9]
 
-# CHECK: stz2g [x1], #-4096
-# CHECK: stz2g [x2], #4080
-# CHECK: stz2g [sp], #16
+# CHECK: stz2g x0,  [x1], #-4096
+# CHECK: stz2g x1,  [x2], #4080
+# CHECK: stz2g x2,  [sp], #16
+# CHECK: stz2g sp,  [sp], #16
 
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x3f,0x04,0xf0,0xd9]
+# NOMTE-NEXT: [0x20,0x04,0xf0,0xd9]
+# NOMTE:      warning: invalid instruction encoding
+# NOMTE-NEXT: [0x41,0xf4,0xef,0xd9]
 # NOMTE:      warning: invalid instruction encoding
-# NOMTE-NEXT: [0x5f,0xf4,0xef,0xd9]
+# NOMTE-NEXT: [0xe2,0x17,0xe0,0xd9]
 # NOMTE:      warning: invalid instruction encoding
 # NOMTE-NEXT: [0xff,0x17,0xe0,0xd9]
 
 [0x40,0x04,0x20,0x69]
 [0x40,0x84,0x1f,0x69]
 [0xe0,0x87,0x00,0x69]
 [0x5f,0x84,0x00,0x69]
 [0x40,0xfc,0x00,0x69]