From b5f4cf63a8b952731053a0d04af0fc8c946d86f3 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Tue, 2 Apr 2024 11:51:36 -0700
Subject: Allow enum values to be used as generic arguments. (#3874)

* Allow enum values to be used as generic arguments.

* Fix constant folding.
---
 .../language-feature/enums/enum-generic-arg.slang  | 33 ++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 tests/language-feature/enums/enum-generic-arg.slang

(limited to 'tests')

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>();
+}
-- 
cgit v1.2.3