[mlir][dxsa] Add dcl_input and dcl_output instructions

Example:
  dxsa.dcl_input %v0
  dxsa.dcl_output %o0

Signed-off-by: Vladimir Shiryaev <tagolog@users.noreply.github.com>

Author: tagolog

mlir/include/mlir/Dialect/DXSA/IR/DXSAOps.td

@@ -534,4 +534,41 @@ def DXSA_DclInputPsSgv : DXSA_Op<"dcl_input_ps_sgv"> {
   let assemblyFormat = "$operand `,` $name attr-dict";
 }
 
+def DXSA_DclInput : DXSA_Op<"dcl_input"> {
+  let summary = "declares a shader input register";
+  let description = [{
+    The `dxsa.dcl_input` operation declares a shader input register.
+
+    The register operand can be either an indexed input register
+    or a special scalar input.
+
+    Example:
+
+    ```mlir
+
+    dxsa.dcl_input %v0
+    ```
+  }];
+  let arguments = (ins DXSA_OperandType:$operand);
+  let assemblyFormat = "$operand attr-dict";
+}
+
+def DXSA_DclOutput : DXSA_Op<"dcl_output"> {
+  let summary = "declares a shader output register";
+  let description = [{
+    The `dxsa.dcl_output` operation declares a shader output register.
+
+    The register operand can be either an indexed output register
+    or a special scalar output.
+
+    Example:
+
+    ```mlir
+    dxsa.dcl_output %o0
+    ```
+  }];
+  let arguments = (ins DXSA_OperandType:$operand);
+  let assemblyFormat = "$operand attr-dict";
+}
+
 #endif // DXSA_OPS

mlir/lib/Target/DXSA/BinaryParser.cpp

@@ -597,6 +597,14 @@ class DXBuilder {
                                        systemValueNameAttr);
   }
 
+  Instruction buildDclInput(Operand operand, Location loc) {
+    return dxsa::DclInput::create(builder, loc, operand);
+  }
+
+  Instruction buildDclOutput(Operand operand, Location loc) {
+    return dxsa::DclOutput::create(builder, loc, operand);
+  }
+
 private:
   MLIRContext *context;
   ModuleOp module;
@@ -1076,6 +1084,20 @@ class Parser {
     return builder.buildDclInputPsSgv(*operand, *systemValueName, loc);
   }
 
+  FailureOr<Instruction> parseDclInput(Location loc) {
+    auto operand = parseOperand();
+    if (failed(operand))
+      return failure();
+    return builder.buildDclInput(*operand, loc);
+  }
+
+  FailureOr<Instruction> parseDclOutput(Location loc) {
+    auto operand = parseOperand();
+    if (failed(operand))
+      return failure();
+    return builder.buildDclOutput(*operand, loc);
+  }
+
   OptionalParseResult parseDclInstruction(uint32_t opcodeToken, Location loc,
                                           Instruction &out) {
     FailureOr<Instruction> result;
@@ -1116,6 +1138,12 @@ class Parser {
     case D3D10_SB_OPCODE_DCL_INPUT_PS_SGV:
       result = parseDclInputPsSgv(loc);
       break;
+    case D3D10_SB_OPCODE_DCL_INPUT:
+      result = parseDclInput(loc);
+      break;
+    case D3D10_SB_OPCODE_DCL_OUTPUT:
+      result = parseDclOutput(loc);
+      break;
     default:
       return std::nullopt;
     }

mlir/test/Target/DXSA/dcl_input.mlir

@@ -0,0 +1,13 @@
+// RUN: mlir-translate --import-dxsa-bin %S/inputs/dcl_input.bin | FileCheck %s
+
+// CHECK-LABEL: module
+module {
+  // dcl_input v0.x
+  // CHECK:      %0 = dxsa.index.imm {imm = 0 : i32}
+  // CHECK-NEXT: %1 = dxsa.operand %0 {mask = 16 : i32, num_components = 4 : i32, type = 1 : i32}
+  // CHECK-NEXT: dxsa.dcl_input %1
+
+  // dcl_input vOutputControlPointID
+  // CHECK-NEXT: %2 = dxsa.operand {num_components = 1 : i32, type = 22 : i32}
+  // CHECK-NEXT: dxsa.dcl_input %2
+}

mlir/test/Target/DXSA/dcl_output.mlir

@@ -0,0 +1,13 @@
+// RUN: mlir-translate --import-dxsa-bin %S/inputs/dcl_output.bin | FileCheck %s
+
+// CHECK-LABEL: module
+module {
+  // dcl_output o0.xyzw
+  // CHECK:      %0 = dxsa.index.imm {imm = 0 : i32}
+  // CHECK-NEXT: %1 = dxsa.operand %0 {mask = 240 : i32, num_components = 4 : i32, type = 2 : i32}
+  // CHECK-NEXT: dxsa.dcl_output %1
+
+  // dcl_output oDepth
+  // CHECK-NEXT: %2 = dxsa.operand {num_components = 1 : i32, type = 12 : i32}
+  // CHECK-NEXT: dxsa.dcl_output %2
+}