[mlir][dxsa] Add dcl_stream instruction

Example:
 dxsa.dcl_stream 0

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

Author: tagolog

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

@@ -969,4 +969,22 @@ def DXSA_DclConstantBuffer : DXSA_Op<"dcl_constant_buffer"> {
   let hasVerifier = 1;
 }
 
+def DXSA_DclStream : DXSA_Op<"dcl_stream"> {
+  let summary = "declares a geometry shader (GS) output stream";
+  let description = [{
+    The `dxsa.dcl_stream` operation declares a geometry shader (GS) output stream.
+
+    The `$index` must be in the range [0, 3].
+
+    Example:
+
+    ```mlir
+    dxsa.dcl_stream 0
+    ```
+  }];
+  let arguments = (ins ConfinedAttr<I32Attr,
+      [IntNonNegative, IntMaxValue<3>]>:$index);
+  let assemblyFormat = [{ $index attr-dict }];
+}
+
 #endif // DXSA_OPS

mlir/lib/Target/DXSA/BinaryParser.cpp

@@ -595,6 +595,11 @@ class DXBuilder {
     return dxsa::DclMaxOutputVertexCount::create(builder, loc, count);
   }
 
+  Instruction buildDclStream(uint32_t index, Location loc) {
+    return dxsa::DclStream::create(builder, loc,
+                                   builder.getI32IntegerAttr(index));
+  }
+
   Instruction buildDclInputPs(dxsa::InterpolationMode interpolationMode,
                               Operand operand, Location loc) {
     auto interpolationModeAttr = dxsa::InterpolationModeAttr::get(
@@ -1144,6 +1149,38 @@ class Parser {
     return builder.buildDclMaxOutputVertexCount(count, loc);
   }
 
+  FailureOr<Instruction> parseDclStream(Location loc) {
+    auto operandToken = parseToken();
+    FAILURE_IF_FAILED(operandToken);
+
+    auto operandType = DECODE_D3D10_SB_OPERAND_TYPE(*operandToken);
+    if (operandType != D3D11_SB_OPERAND_TYPE_STREAM)
+      return emitError(loc, "unexpected operand type: ") << operandType;
+
+    if (DECODE_IS_D3D10_SB_OPERAND_EXTENDED(*operandToken))
+      return emitError(loc, "extended operand tokens are not supported");
+
+    auto numComponents = DECODE_D3D10_SB_OPERAND_NUM_COMPONENTS(*operandToken);
+    if (numComponents != D3D10_SB_OPERAND_0_COMPONENT)
+      return emitError(loc, "unexpected number of components: ")
+             << numComponents;
+
+    auto indexDim = DECODE_D3D10_SB_OPERAND_INDEX_DIMENSION(*operandToken);
+    if (indexDim != D3D10_SB_OPERAND_INDEX_1D)
+      return emitError(loc, "unsupported index dimension: ") << indexDim;
+
+    auto indexRepresentation =
+        DECODE_D3D10_SB_OPERAND_INDEX_REPRESENTATION(0, *operandToken);
+    if (indexRepresentation != D3D10_SB_OPERAND_INDEX_IMMEDIATE32)
+      return emitError(loc, "unsupported index representation: ")
+             << indexRepresentation;
+
+    auto indexToken = parseToken();
+    FAILURE_IF_FAILED(indexToken);
+
+    return builder.buildDclStream(*indexToken, loc);
+  }
+
   FailureOr<dxsa::InterpolationMode>
   parseInterpolationMode(uint32_t opcodeToken, Location loc) {
     auto rawInterpolationMode =
@@ -1423,6 +1460,9 @@ class Parser {
     case D3D10_SB_OPCODE_DCL_MAX_OUTPUT_VERTEX_COUNT:
       result = parseDclMaxOutputVertexCount(loc);
       break;
+    case D3D11_SB_OPCODE_DCL_STREAM:
+      result = parseDclStream(loc);
+      break;
     case D3D10_SB_OPCODE_DCL_INPUT_PS:
       result = parseDclInputPs(opcodeToken, loc);
       break;

mlir/test/Target/DXSA/dcl_stream.mlir

@@ -0,0 +1,6 @@
+// RUN: mlir-translate --import-dxsa-bin %S/inputs/dcl_stream.bin | FileCheck %s
+
+// CHECK:      module {
+// CHECK-NEXT:   dxsa.dcl_stream 0
+// CHECK-NEXT:   dxsa.dcl_stream 3
+// CHECK-NEXT: }

mlir/test/Target/DXSA/dcl_stream_invalid.mlir

@@ -0,0 +1,9 @@
+// RUN: mlir-opt %s -split-input-file -verify-diagnostics
+
+// expected-error@+1 {{attribute 'index' failed to satisfy constraint: 32-bit signless integer attribute whose value is non-negative whose maximum value is 3}}
+dxsa.dcl_stream -1
+
+// -----
+
+// expected-error@+1 {{attribute 'index' failed to satisfy constraint: 32-bit signless integer attribute whose value is non-negative whose maximum value is 3}}
+dxsa.dcl_stream 4