Add datalayout for constant buffers. (#5608)

* Add datalayout for constant buffers.

* Fixes.

* Fix test.

* Fix glsl codegen.

* Update spirv-specific doc.

* Fix test.

* Fix binding in the presense of specialization constants.

* address comments.

* Add a test for constant buffer layout.

1 files changed, 45 insertions, 0 deletions

diff --git a/tests/spirv/constant-buffer-layout.slang b/tests/spirv/constant-buffer-layout.slang
new file mode 100644
index 000000000..90aa1eef8
--- /dev/null
+++ b/tests/spirv/constant-buffer-layout.slang
@@ -0,0 +1,45 @@
+//TEST:SIMPLE(filecheck=SPIRV): -target spirv -emit-spirv-directly
+
+//SPIRV: ArrayStride 12
+
+struct Test
+{
+//SPIRV: Offset 0
+    uint v0;
+
+//SPIRV: Offset 4
+// matrix always start on a new register
+    float3x3 v1;
+//SPIRV: Offset 40
+// Non-matrix can pack with a partially filled register
+    uint v2;
+};
+
+ConstantBuffer<Test, ScalarDataLayout> buffer;
+
+//TEST_INPUT:ubuffer(data=[0], stride=4):out,name outputBuffer
+RWStructuredBuffer<int> outputBuffer;
+
+__generic<T : IArithmetic, let N : int>
+bool comp(vector<T,N> v1, vector<T,N> v2)
+{
+    for (uint i = 0; i < N; i++)
+        if (v1[i] != v2[i])
+            return false;
+
+    return true;
+}
+
+[shader("compute")]
+[numthreads(2, 2, 1)]
+void computeMain()
+{
+    // CHECK: 64
+    outputBuffer[0] = (true
+            && buffer.v0 == 1
+            && comp(buffer.v1[0], float3(2, 3, 4))
+            && comp(buffer.v1[1], float3(5, 6, 7))
+            && comp(buffer.v1[2], float3(8, 9, 10))
+            && buffer.v2 == 11
+        ) ? 100 : 0;
+}
\ No newline at end of file