From 3fa382505271834514d47612efee8e51a06204c5 Mon Sep 17 00:00:00 2001
From: jarcherNV <jarcher@nvidia.com>
Date: Tue, 10 Jun 2025 09:44:08 -0700
Subject: Allow checking capabilities in specific stages (#7375)

This allows checking capabilities in any stage, needed specifically for
the hlsl_2018 capability which is defined for sm_5_1 and above. Stage
specific capabilities such as cs_5_1 would not find this in any stage
other than compute, so we need to restrict the check to only desired
stages.
---
 tests/hlsl/hlsl-capability.slang | 49 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 tests/hlsl/hlsl-capability.slang

(limited to 'tests')

diff --git a/tests/hlsl/hlsl-capability.slang b/tests/hlsl/hlsl-capability.slang
new file mode 100644
index 000000000..3a950034e
--- /dev/null
+++ b/tests/hlsl/hlsl-capability.slang
@@ -0,0 +1,49 @@
+//TEST:SIMPLE(filecheck=CHECK_CS): -target hlsl -stage compute -entry computeMain -profile cs_6_3
+//TEST:SIMPLE(filecheck=CHECK_SM): -target hlsl -stage compute -entry computeMain -profile sm_6_3
+//TEST:SIMPLE(filecheck=CHECK_CS_CAP): -target hlsl -stage compute -entry computeMain -profile cs_6_3 -capability hlsl_2018
+//TEST:SIMPLE(filecheck=CHECK_SM_CAP): -target hlsl -stage compute -entry computeMain -profile sm_6_3 -capability hlsl_2018
+
+// Test IR code generation for the `?:` "select" operator with the hlsl_2018 capability and cs_6_3 profile.
+
+// Verify that select is emitted for cs_6_3 and sm_6_3.
+// CHECK_CS: select({{.*}})
+// CHECK_SM: select({{.*}})
+// CHECK_CS-NOT: {{.*}}?{{.*}}:{{.*}}
+// CHECK_SM-NOT: {{.*}}?{{.*}}:{{.*}}
+
+// Verify that select is not emitted for cs_6_3 and sm_6_3 with the hlsl_2018 capability.
+// CHECK_CS_CAP-NOT: select({{.*}})
+// CHECK_SM_CAP-NOT: select({{.*}})
+// CHECK_CS_CAP: {{.*}}?{{.*}}:{{.*}}
+// CHECK_SM_CAP: {{.*}}?{{.*}}:{{.*}}
+
+RWStructuredBuffer<int> outputBuffer;
+static int result = 0;
+bool2 assignFunc(int index)
+{
+    result++;
+    return bool2(true);
+}
+
+[numthreads(4, 1, 1)]
+void computeMain(int3 dispatchThreadID : SV_DispatchThreadID)
+{
+    int index = dispatchThreadID.x;
+
+    if (all(bool2(index >= 1) && assignFunc(index)))
+    {
+        result++;
+    }
+
+    if (all(bool2(index >= 2) || !assignFunc(index)))
+    {
+        result++;
+    }
+
+    if (all(bool2(index >= 3) ? assignFunc(index) : bool2(false)))
+    {
+        result++;
+    }
+
+    outputBuffer[index] = result;
+}
-- 
cgit v1.2.3