From 2c4f9810327d58023e9ec44f579cd78adf56317b Mon Sep 17 00:00:00 2001 From: Yong He Date: Mon, 1 Apr 2024 15:56:02 -0700 Subject: Allow bit operators on enum types. (#3862) * Allow bit operators on enum types. * Fix. --- tests/language-feature/enums/enum-bit-ops.slang | 32 +++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 tests/language-feature/enums/enum-bit-ops.slang (limited to 'tests/language-feature') diff --git a/tests/language-feature/enums/enum-bit-ops.slang b/tests/language-feature/enums/enum-bit-ops.slang new file mode 100644 index 000000000..abffe8dee --- /dev/null +++ b/tests/language-feature/enums/enum-bit-ops.slang @@ -0,0 +1,32 @@ +//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -shaderobj + +[Flags] +enum MyFlags +{ + Zero = 0, + BitOne, // = 1 + BitTwo, // = 2 + BitThree // = 4 +} + +bool test(MyFlags f1) +{ + return (f1 & MyFlags.BitTwo) != 0; +} + +//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name=outputBuffer +RWStructuredBuffer outputBuffer; + +[numthreads(1, 1, 1)] +void computeMain(int3 dispatchThreadID: SV_DispatchThreadID) +{ + int outVal = test(MyFlags.BitOne | MyFlags.BitTwo) ? 1 : 0; + // CHECK: 1 + outputBuffer[0] = outVal; + // CHECK: 4 + outputBuffer[1] = (int)MyFlags.BitThree; + // CHECK: 0 + outputBuffer[2] = test(MyFlags.BitOne | MyFlags.BitThree) ? 1 : 0; + // CHECK: 8 + outputBuffer[3] = MyFlags(8); +} -- cgit v1.2.3