Fix invalid code generation for when using nested resource specialization (#4751)

3 files changed, 118 insertions, 0 deletions

diff --git a/tests/language-feature/resource-specialization-nested-specialization.slang b/tests/language-feature/resource-specialization-nested-specialization.slang
new file mode 100644
index 000000000..2b88e7611
--- /dev/null
+++ b/tests/language-feature/resource-specialization-nested-specialization.slang
@@ -0,0 +1,58 @@
+//TEST:SIMPLE(filecheck=CHECK_DXIL): -target dxil -profile sm_6_0 -entry computeMain -stage compute -DMEMBER_FUNCTION_CALL
+//TEST:SIMPLE(filecheck=CHECK_DXIL): -target dxil -profile sm_6_0 -entry computeMain -stage compute
+
+//CHECK_DXIL: computeMain
+
+struct Grid
+{
+    uint bufSize;
+    StructuredBuffer<uint> buf;
+};
+
+struct GridGeo
+{
+    Grid grids[2];
+
+    void getGrid(uint index, out Grid grid)
+    {
+        grid = grids[index];
+    }
+};
+
+struct Scene
+{
+    GridGeo gridGeo;
+
+    void getGrid_BAD(uint index, out Grid grid)
+    {
+        gridGeo.getGrid(index, grid);
+    }
+
+    void getGrid_GOOD(uint index, out Grid grid)
+    {
+        grid = gridGeo.grids[index];
+    }
+};
+
+ParameterBlock<Scene> gScene;
+RWStructuredBuffer<uint> gridBuffers[2];
+RWStructuredBuffer<uint> outputBuffer;
+
+void direct_getGrid_BAD(uint index, out Grid grid)
+{
+    gScene.gridGeo.getGrid(index, grid);
+}
+
+[numthreads(1, 1, 1)]
+void computeMain(uint3 threadId: SV_DispatchThreadID)
+{
+   
+    Grid grid;
+
+#ifdef MEMBER_FUNCTION_CALL
+    direct_getGrid_BAD(1, grid);
+#else 
+    gScene.getGrid_BAD(1, grid);
+#endif
+    gridBuffers[0][1] = grid.buf[1];
+}
diff --git a/tests/language-feature/resource-specialization-struct-out.slang b/tests/language-feature/resource-specialization-struct-out.slang
new file mode 100644
index 000000000..b525dafb9
--- /dev/null
+++ b/tests/language-feature/resource-specialization-struct-out.slang
@@ -0,0 +1,31 @@
+//TEST:SIMPLE(filecheck=CHECK_DXIL):-target dxil -entry computeMain -profile cs_6_2
+//CHECK_DXIL: computeMain
+
+// This test demonstrates out parameter with a struct & resource type.
+    
+RWTexture1D<int> g_t;
+
+RWStructuredBuffer<int> outputBuffer;
+
+struct Thing
+{
+    int a;
+    RWTexture1D<int> t;
+};
+
+void setThing(out Thing t)
+{
+    t.a = 10;
+    t.t = g_t;
+}
+
+[numthreads(4, 4, 1)]
+void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+    int x = dispatchThreadID.x;
+   
+    Thing thing;
+    setThing(thing);
+        
+    outputBuffer[dispatchThreadID.x] = x + thing.t.Load(1); 
+}
diff --git a/tests/language-feature/resource-specialization-struct-return.slang b/tests/language-feature/resource-specialization-struct-return.slang
new file mode 100644
index 000000000..ca01f55d2
--- /dev/null
+++ b/tests/language-feature/resource-specialization-struct-return.slang
@@ -0,0 +1,29 @@
+//TEST:SIMPLE(filecheck=CHECK_DXIL):-target dxil -entry computeMain -profile cs_6_2
+//CHECK_DXIL: computeMain
+
+// This test demonstrates returning struct with resource.
+
+RWTexture1D<int> g_t;
+RWStructuredBuffer<int> outputBuffer;
+
+struct Thing
+{
+    int a;
+    RWTexture1D<int> t;
+};
+
+Thing makeThing()
+{
+ 	Thing t;
+    t.a = 10;
+    t.t = g_t;
+    return t;
+}
+
+[numthreads(4, 4, 1)]
+void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+    int x = dispatchThreadID.x;
+    Thing thing = makeThing();
+    outputBuffer[dispatchThreadID.x] = x + thing.t.Load(1); 
+}
\ No newline at end of file