Allow requiring glsl language extensions on structs (#6173)

* Allow requiring glsl language extensions on structs

* format code

---------

Co-authored-by: slangbot <186143334+slangbot@users.noreply.github.com>
Co-authored-by: Yong He <yonghe@outlook.com>

1 files changed, 27 insertions, 0 deletions

diff --git a/tests/language-feature/glsl-extension.slang b/tests/language-feature/glsl-extension.slang
new file mode 100644
index 000000000..dbf8fe86c
--- /dev/null
+++ b/tests/language-feature/glsl-extension.slang
@@ -0,0 +1,27 @@
+//TEST:SIMPLE(filecheck=CHECK): -target glsl -entry main -stage compute
+
+RWStructuredBuffer<float> outputBuffer;
+
+// Check that extensions on types work
+__glsl_extension(GL_BAR_bar)
+// CHECK-DAG: #extension GL_BAR_bar : require
+struct S
+{
+    float a;
+}
+
+// Check that extensions on functions work
+__glsl_extension(GL_FOO_foo)
+// CHECK-DAG: #extension GL_FOO_foo : require
+float foo(S s)
+{
+    return s.a;
+}
+
+[numthreads(4, 1, 1)]
+void main(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+    S s;
+    s.a = 0;
+    outputBuffer[dispatchThreadID.x] = foo(s);
+}