2 files changed, 43 insertions, 0 deletions
diff --git a/tests/compute/multiple-continue-sites.slang b/tests/compute/multiple-continue-sites.slang
new file mode 100644
index 000000000..5a5b78d0c
--- /dev/null
+++ b/tests/compute/multiple-continue-sites.slang
@@ -0,0 +1,39 @@
+//TEST(compute):COMPARE_COMPUTE:
+//TEST_INPUT:ubuffer(data=[0 1 2 3], stride=4):dxbinding(0),glbinding(0),out
+
+// Test that a loop with multiple `continue` sites works.
+//
+// The current Slang codegen strategy for `continue` ends
+// up duplicating the "continue clause" for a `for` loop
+// at each `continue` site, so it will stress-test any
+// code that assumes a given instruction/block only
+// appears once in the region tree.
+//
+
+int test(int inVal)
+{
+	int ii = inVal;
+	for(;!(ii & 0x20); ii += 0x10)
+	{
+		if(ii == 2)
+		{
+			continue;
+		}
+
+		ii += 0x100;
+		// there is an implicit `continue` here
+	}
+
+	return ii;
+}
+
+RWStructuredBuffer<int> outputBuffer : register(u0);
+
+[numthreads(4, 1, 1)]
+void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+	uint tid = dispatchThreadID.x;
+	int inVal = outputBuffer[tid];
+	int outVal = test(inVal);
+	outputBuffer[tid] = outVal;
+}
+\ No newline at end of file
diff --git a/tests/compute/multiple-continue-sites.slang.expected.txt b/tests/compute/multiple-continue-sites.slang.expected.txt
new file mode 100644
index 000000000..6752a80e0
--- /dev/null
+++ b/tests/compute/multiple-continue-sites.slang.expected.txt
@@ -0,0 +1,4 @@
+220
+221
+122
+223