[mlir][dxsa] Add dcl_resource_structured instruction

Example:
  dxsa.dcl_resource_structured <type = resource, components = 0, index = [0, 0, 3]>, <struct_byte_stride = 32, space = 1>

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

Author: tagolog

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

@@ -922,4 +922,35 @@ def DXSA_DclTgsmStructured : DXSA_Op<"dcl_tgsm_structured"> {
   let hasVerifier = 1;
 }
 
+def DXSA_DclResourceStructured : DXSA_Op<"dcl_resource_structured"> {
+  let summary = "declares a structured buffer shader input resource bound to a register";
+  let description = [{
+    The `dxsa.dcl_resource_structured` operation declares a structured buffer
+    shader input resource bound to a register.
+
+    `$struct_byte_stride` is the structure size in bytes; it must be a multiple of 4.
+
+    Example:
+
+    ```mlir
+    dxsa.dcl_resource_structured <type = resource, components = 0, index = [3]>,
+        <struct_byte_stride = 16>
+    dxsa.dcl_resource_structured <type = resource, components = 0, index = [0, 0, 3]>,
+        <struct_byte_stride = 32, space = 1>
+    ```
+  }];
+
+  let arguments = (ins
+      DXSA_InlineOperandAttr:$operand,
+      ConfinedAttr<I32Attr, [IntPositive]>:$struct_byte_stride,
+      OptionalAttr<I32Attr>:$space);
+  let assemblyFormat = [{
+    $operand `,`
+    `<` `struct_byte_stride` `=` $struct_byte_stride
+         (`,` `space` `=` $space^)? `>`
+    attr-dict
+  }];
+  let hasVerifier = 1;
+}
+
 #endif // DXSA_OPS

mlir/lib/Dialect/DXSA/IR/DXSA.cpp

@@ -84,6 +84,18 @@ LogicalResult DclTgsmStructured::verify() {
   return success();
 }
 
+LogicalResult DclResourceStructured::verify() {
+  auto operandType = getOperand().getType();
+  if (operandType != InlineOperandType::resource)
+    return emitOpError("operand must be a resource register, got ")
+           << stringifyInlineOperandType(operandType);
+  auto stride = getStructByteStride();
+  if (stride % 4 != 0)
+    return emitOpError("struct byte stride must be a multiple of 4, got ")
+           << stride;
+  return success();
+}
+
 //===----------------------------------------------------------------------===//
 // TableGen'd attribute method definitions
 //===----------------------------------------------------------------------===//

mlir/lib/Target/DXSA/BinaryParser.cpp

@@ -692,6 +692,15 @@ class DXBuilder {
         builder.getI32IntegerAttr(structCount));
   }
 
+  Instruction buildDclResourceStructured(dxsa::InlineOperandAttr operand,
+                                         uint32_t structByteStride,
+                                         std::optional<uint32_t> space,
+                                         Location loc) {
+    auto spaceAttr = space ? builder.getI32IntegerAttr(*space) : IntegerAttr();
+    return dxsa::DclResourceStructured::create(builder, loc, operand,
+                                               structByteStride, spaceAttr);
+  }
+
 private:
   MLIRContext *context;
   ModuleOp module;
@@ -1317,6 +1326,25 @@ class Parser {
                                           *structCount, loc);
   }
 
+  FailureOr<Instruction> parseDclResourceStructured(Location loc) {
+    auto operand = parseInlineOperand();
+    FAILURE_IF_FAILED(operand);
+
+    auto strideToken = parseToken();
+    FAILURE_IF_FAILED(strideToken);
+
+    std::optional<uint32_t> space;
+    auto indexArray = operand->getIndex();
+    if (indexArray && indexArray.size() == 3) {
+      auto spaceToken = parseToken();
+      FAILURE_IF_FAILED(spaceToken);
+      space = *spaceToken;
+    }
+
+    return builder.buildDclResourceStructured(*operand, *strideToken, space,
+                                              loc);
+  }
+
   OptionalParseResult parseDclInstruction(uint32_t opcodeToken, Location loc,
                                           Instruction &out) {
     FailureOr<Instruction> result;
@@ -1393,6 +1421,9 @@ class Parser {
     case D3D11_SB_OPCODE_DCL_THREAD_GROUP_SHARED_MEMORY_STRUCTURED:
       result = parseDclTgsmStructured(loc);
       break;
+    case D3D11_SB_OPCODE_DCL_RESOURCE_STRUCTURED:
+      result = parseDclResourceStructured(loc);
+      break;
     default:
       return std::nullopt;
     }

mlir/test/Target/DXSA/dcl_resource_structured.mlir

@@ -0,0 +1,6 @@
+// RUN: mlir-translate --import-dxsa-bin %S/inputs/dcl_resource_structured.bin | FileCheck %s
+
+// CHECK:      module {
+// CHECK-NEXT:   dxsa.dcl_resource_structured <type = resource, components = 0, index = [3]>, <struct_byte_stride = 16>
+// CHECK-NEXT:   dxsa.dcl_resource_structured <type = resource, components = 0, index = [0, 0, 3]>, <struct_byte_stride = 32, space = 1>
+// CHECK-NEXT: }

mlir/test/Target/DXSA/dcl_resource_structured_invalid.mlir

@@ -0,0 +1,14 @@
+// RUN: mlir-opt %s -split-input-file -verify-diagnostics
+
+// expected-error@+1 {{'dxsa.dcl_resource_structured' op operand must be a resource register, got temp}}
+dxsa.dcl_resource_structured <type = temp, components = 0, index = [0]>, <struct_byte_stride = 16>
+
+// -----
+
+// expected-error@+1 {{'dxsa.dcl_resource_structured' op struct byte stride must be a multiple of 4, got 6}}
+dxsa.dcl_resource_structured <type = resource, components = 0, index = [0]>, <struct_byte_stride = 6>
+
+// -----
+
+// expected-error@+1 {{attribute 'struct_byte_stride' failed to satisfy constraint: 32-bit signless integer attribute whose value is positive}}
+dxsa.dcl_resource_structured <type = resource, components = 0, index = [0]>, <struct_byte_stride = 0>