Add support for resource arrays to the offload test suite (#302)

Introduces support for resource arrays across the offload test suite, including:
- Data input and output handling
- Comparison of expected results
- Integration with both DirectX and Vulkan runtimes (Metal is not
supported yet)

A new property `ArraySize` has been added to the `Buffer` definition. If
`ArraySize` is `1` (the default), the `Data` field of the `Buffer`
remains unchanged:

```
Data: [0, 1, 2, 3]
```

If `ArraySize` is greater than `1`, the `Data` field must be specified
as a list of arrays:
```
Data:
  - [0, 1, 2, 3]
  - [4, 5, 6, 7]
  - [8, 9, 10, 11]
```

If the resource array includes resources with counters, their values
will be output using the `Counters` field:
```
Counters: [1, 2, 3]
```

Closes llvm/wg-hlsl#291

Author: hekota

include/Image/Image.h

@@ -90,7 +90,10 @@ class ImageRef {
                  B.getSingleElementSize(), B.Channels,
                  B.Format == DataFormat::Float32 ||
                      B.Format == DataFormat::Float64,
-                 llvm::StringRef(B.Data.get(), B.size())) {}
+                 llvm::StringRef(B.Data.back().get(), B.size())) {
+    assert(B.ArraySize == 1 &&
+           "Buffer must not be an array to be used for ImageRef");
+  }
 
   uint32_t getHeight() const { return Height; }
   uint32_t getWidth() const { return Width; }

include/Support/Pipeline.h

@@ -78,10 +78,15 @@ struct Buffer {
   DataFormat Format;
   int Channels;
   int Stride;
-  std::unique_ptr<char[]> Data;
+  uint32_t ArraySize;
+  // Data can contain one block of data for a singular resource
+  // or multiple blocks for a resource array.
+  llvm::SmallVector<std::unique_ptr<char[]>> Data;
   size_t Size;
   OutputProperties OutputProps;
-  uint32_t Counter;
+  // Counters can contain one counter value for a singular resource
+  // or multiple values for an array of resources with counters.
+  llvm::SmallVector<uint32_t> Counters;
 
   uint32_t size() const { return Size; }
 
@@ -267,6 +272,14 @@ struct Pipeline {
     return DescriptorCount;
   }
 
+  uint32_t getDescriptorCountWithFlattenedArrays() const {
+    uint32_t DescriptorCount = 0;
+    for (auto &D : Sets)
+      for (auto &R : D.Resources)
+        DescriptorCount += R.BufferPtr->ArraySize;
+    return DescriptorCount;
+  }
+
   Buffer *getBuffer(llvm::StringRef Name) {
     for (auto &B : Buffers)
       if (Name == B.Name)
@@ -412,6 +425,32 @@ template <> struct ScalarEnumerationTraits<offloadtest::dx::RootParamKind> {
 #undef ENUM_CASE
   }
 };
+
+template <typename T> struct SequenceTraits<SmallVector<SmallVector<T>>> {
+  static size_t size(IO &io, SmallVector<SmallVector<T>> &seq) {
+    return seq.size();
+  }
+
+  static SmallVector<T> &element(IO &io, SmallVector<SmallVector<T>> &seq,
+                                 size_t index) {
+    if (index >= seq.size())
+      seq.resize(index + 1);
+    return seq[index];
+  }
+};
+
+template <typename T> struct SequenceTraits<SmallVector<MutableArrayRef<T>>> {
+  static size_t size(IO &io, SmallVector<MutableArrayRef<T>> &seq) {
+    return seq.size();
+  }
+
+  static MutableArrayRef<T> &
+  element(IO &io, SmallVector<MutableArrayRef<T>> &seq, size_t index) {
+    assert(index < seq.size());
+    return seq[index];
+  }
+};
+
 } // namespace yaml
 } // namespace llvm