From fdf061e278720ec066a1fac8f1f35a22e817bf2d Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Thu, 21 Nov 2024 14:07:23 -0800
Subject: 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.
---
 tests/spirv/constant-buffer-layout.slang | 45 ++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 tests/spirv/constant-buffer-layout.slang

(limited to 'tests/spirv/constant-buffer-layout.slang')

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
-- 
cgit v1.2.3