2 files changed, 40 insertions, 0 deletions
diff --git a/tests/glsl/fragment-invocation-density-glsl.slang b/tests/glsl/fragment-invocation-density-glsl.slang
new file mode 100644
index 000000000..7cee4620b
--- /dev/null
+++ b/tests/glsl/fragment-invocation-density-glsl.slang
@@ -0,0 +1,15 @@
+//TEST:SIMPLE(filecheck=CHECK-GLSL): -target glsl -stage fragment -entry main
+
+// Test for EXT_fragment_invocation_density support (gl_FragSizeEXT and gl_FragInvocationCountEXT )
+import glsl;
+
+layout(location = 0) out highp vec4 FragColor;
+
+void main()
+{
+    // CHECK-GLSL: #extension GL_EXT_fragment_invocation_density : require
+    // CHECK-GLSL-DAG: gl_FragSizeEXT
+    // CHECK-GLSL-DAG: gl_FragInvocationCountEXT
+
+    FragColor = vec4(gl_FragSizeEXT.x, gl_FragSizeEXT.y, gl_FragInvocationCountEXT, 1.0);
+}
+\ No newline at end of file
diff --git a/tests/spirv/fragment-invocation-density.slang b/tests/spirv/fragment-invocation-density.slang
new file mode 100644
index 000000000..f26b403ec
--- /dev/null
+++ b/tests/spirv/fragment-invocation-density.slang
@@ -0,0 +1,25 @@
+//TEST:SIMPLE(filecheck=CHECK-SPIRV): -target spirv -stage fragment -entry main
+//TEST:SIMPLE(filecheck=CHECK-GLSL): -target glsl -stage fragment -entry main
+
+// Test for SPV_EXT_fragment_invocation_density support (SV_FragSize and SV_FragInvocationCount)
+
+struct FragmentInput
+{
+    int2 fragmentSize : SV_FragSize;
+    int invocationsPerPixel : SV_FragInvocationCount;
+}
+
+[shader("fragment")]
+float4 main(FragmentInput input) : SV_Target
+{
+    // CHECK-SPIRV: OpCapability FragmentDensityEXT
+    // CHECK-SPIRV: OpExtension "SPV_EXT_fragment_invocation_density"
+    // CHECK-SPIRV-DAG: OpDecorate {{.*}} BuiltIn FragSizeEXT
+    // CHECK-SPIRV-DAG: OpDecorate {{.*}} BuiltIn FragInvocationCountEXT
+
+    // CHECK-GLSL: #extension GL_EXT_fragment_invocation_density : require
+    // CHECK-GLSL-DAG: gl_FragSizeEXT
+    // CHECK-GLSL-DAG: gl_FragInvocationCountEXT
+
+    return float4(input.fragmentSize.x, input.fragmentSize.y, input.invocationsPerPixel, 1.0);
+}
+\ No newline at end of file