Implement more shifts in tablegen

leadpogrommer · Apr 8, 2024 · fa00369 · fa00369
1 parent 16d1165
commit fa00369
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/llvm/lib/Target/CDM/CDMInstrInfo.td b/llvm/lib/Target/CDM/CDMInstrInfo.td
@@ -99,15 +99,22 @@ def : Pat<(add i16
 // def : Pattern<(i16 (load_sym tglobaladdr:$addr)),  [(ldi $addr)]>;
 
 
-def shamt       : Operand<i16>;
+def shamt: Operand<i16>;
 def immZext3: ImmLeaf<i16, [{return (Imm > 0) && (Imm <= 8);}]>;
-def SHL: CDMInst<(outs CPURegs:$rd), (ins CPURegs:$rs, shamt:$shamt),
-        "shl\t$rs, $rd, $shamt",
-        [(set CPURegs:$rd, (shl CPURegs:$rs, immZext3:$shamt))]
->{
+
+class ShiftImm3<string instr_asm, SDNode OpNode> :
+  CDMInst<(outs CPURegs:$rd), (ins CPURegs:$rs, shamt:$shamt),
+     !strconcat(instr_asm, "\t$rs, $rd, $shamt"),
+     [(set CPURegs:$rd, (OpNode CPURegs:$rs, immZext3:$shamt))]> {
   let Defs = [PSR];
 }
 
+def SHL  : ShiftImm3<"shl", shl>;
+def SHRA : ShiftImm3<"shra", sra>;
+def SHR  : ShiftImm3<"shr", srl>;
+def ROL  : ShiftImm3<"rol", rotl>;
+def ROR  : ShiftImm3<"ror", rotr>;
+
 // load
 class AlignedLoad<PatFrag Node> :
   PatFrag<(ops node:$ptr), (Node node:$ptr), [{