From ebe8ddefc48478307d5f206cd3e40c41d28a36e3 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Tue, 26 Sep 2023 23:56:06 -0700
Subject: Various SPIRV fixes. (#3231)

* Various SPIRV fixes.

- Geometry shader support (WIP).
- Fix texture get dimension and load.
- Fold global GetElement(MakeArray/MakeVector) insts.
- Call spvopt to inline all functions.
- Translate OpImageSubscript.
- Emit struct member names and global variable names.
- Fix lowering of OpBitNot -> OpNot, instead of OpBitReverse.

* Fix test.

* Fix geometry shader.

* Fix geometry shader emit.

* Add atomic Image access test.

* Fix tests.

* don't fail if spirv-opt fails.

* Update comments.

* Fix test.

* Cleanups.

* indentation

---------

Co-authored-by: Yong He <yhe@nvidia.com>
Co-authored-by: Ellie Hermaszewska <ellieh@nvidia.com>
---
 tests/spirv/bit-field.slang | 54 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 tests/spirv/bit-field.slang

(limited to 'tests/spirv/bit-field.slang')

diff --git a/tests/spirv/bit-field.slang b/tests/spirv/bit-field.slang
new file mode 100644
index 000000000..878b4483d
--- /dev/null
+++ b/tests/spirv/bit-field.slang
@@ -0,0 +1,54 @@
+// bit-field.slang
+
+//TEST(compute, vulkan):COMPARE_COMPUTE_EX(filecheck-buffer=BUFFER):-vk -compute -output-using-type
+
+// Test direct SPIR-V emit on bit field operations.
+
+//TEST_INPUT:set resultBuffer = out ubuffer(data=[0 0 0 0], stride=4)
+RWStructuredBuffer<uint> resultBuffer;
+
+enum Flag
+{
+    AAA = 1,
+    BBB = 2,
+    CCC = 4,
+}
+struct Data
+{
+    uint flags;
+    [mutating] void setFlag(Flag g, bool v)
+    {
+        const uint bit = uint(g) << 8;
+        if (v) flags |= bit;
+        else flags &= ~bit;
+    }
+    bool hasFlag(Flag g)
+    {
+        const uint bit = uint(g) << 8;
+        return (flags & bit) != 0;
+    }
+}
+
+//TEST_INPUT: set Uniforms.d = new Data{0};
+cbuffer Uniforms
+{
+    Data d;
+}
+
+// BUFFER: 1
+// BUFFER: 1
+// BUFFER: 1
+// BUFFER: 0
+
+[numthreads(1,1,1)]
+void computeMain(uint3 dispatchThreadID : SV_DispatchThreadID)
+{
+    Data d1 = d;
+    d1.setFlag(Flag.BBB, true);
+    d1.setFlag(Flag.AAA, true);
+    resultBuffer[0] = d1.hasFlag(Flag.AAA) ? 1 : 0;
+    resultBuffer[1] = d1.hasFlag(Flag.BBB) ? 1 : 0;
+    d1.setFlag(Flag.BBB, false);
+    resultBuffer[2] = d1.hasFlag(Flag.AAA) ? 1 : 0;
+    resultBuffer[3] = d1.hasFlag(Flag.BBB) ? 1 : 0;
+}
-- 
cgit v1.2.3