From ad45062fcc76cd622aaa1a9cb00515d42f8d3528 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Tue, 23 Jan 2024 19:37:10 -0800
Subject: SPIRV Legalization fixes. (#3479)

* Fix CFG legalization for SPIRV backend.

* Emit DepthReplacing execution mode.

* Fix do-while lowering.

---------

Co-authored-by: Yong He <yhe@nvidia.com>
---
 tests/spirv/while-continue.slang | 45 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 tests/spirv/while-continue.slang

(limited to 'tests/spirv/while-continue.slang')

diff --git a/tests/spirv/while-continue.slang b/tests/spirv/while-continue.slang
new file mode 100644
index 000000000..9c7b89295
--- /dev/null
+++ b/tests/spirv/while-continue.slang
@@ -0,0 +1,45 @@
+//TEST:SIMPLE(filecheck=CHECK): -target spirv -stage compute -entry main -emit-spirv-directly
+
+// Test that we can transform a continue in a while loop to valid spirv.
+
+uniform int loopCount;
+
+// CHECK: OpEntryPoint
+
+bool condition1()
+{
+    AllMemoryBarrier();
+    return output[2] < 1;
+}
+bool condition2()
+{
+    AllMemoryBarrier();
+    return output[3] < 1;
+}
+
+RWStructuredBuffer<float> output;
+
+[numthreads(1,1,1)]
+void main()
+{
+    float weight = 0.0;
+    do 
+    {
+        if (condition2()) 
+        { 
+            continue;
+        }
+        AllMemoryBarrier();
+    } while (condition1());
+
+    while(condition1())
+    {
+        if (condition2())
+        {
+            continue;
+        }
+        AllMemoryBarrier();
+    }
+
+    output[0] = weight;
+}
-- 
cgit v1.2.3