Allow enum values to be used as generic arguments. (#3874)

* Allow enum values to be used as generic arguments.

* Fix constant folding.

1 files changed, 33 insertions, 0 deletions

diff --git a/tests/language-feature/enums/enum-generic-arg.slang b/tests/language-feature/enums/enum-generic-arg.slang
new file mode 100644
index 000000000..e851727e1
--- /dev/null
+++ b/tests/language-feature/enums/enum-generic-arg.slang
@@ -0,0 +1,33 @@
+//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -shaderobj
+
+// Test that enum values can be used as compile time constants
+// to specialize generics.
+
+[Flags]
+enum BitFlags
+{
+    One, Two, Three
+}
+
+int test<let F : BitFlags>()
+{
+    return F;
+}
+
+int testInt<let f : int>()
+{
+    return f;
+}
+
+//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name=outputBuffer
+RWStructuredBuffer<int> outputBuffer;
+
+[numthreads(1, 1, 1)]
+void computeMain(int3 dispatchThreadID : SV_DispatchThreadID)
+{
+    // CHECK: 3
+    outputBuffer[0] = test<BitFlags.One | BitFlags.Two>();
+
+    // CHECK: 3
+    outputBuffer[1] = testInt<BitFlags.One | BitFlags.Two>();
+}