[lit] Add dxil version to available_features for lit (microsoft#6310)

This will allow lit test check dxil version to avoid running on
unsupported dxil.dll.

Author: python3kgae

tools/clang/test/CodeGenDXIL/hlsl/objects/NodeObjects/node-objects-metdata.hlsl

@@ -1,8 +1,22 @@
-// RUN: %dxc -T lib_6_8 %s | FileCheck %s --check-prefixes=MD
-// RUN: %dxc -T lib_6_8 -Od %s | FileCheck %s --check-prefixes=MD
-// RUN: %dxc -T lib_6_8 -Zi %s | FileCheck %s --check-prefixes=MD
-
-// RUN: %dxc -T lib_6_8 -fcgl %s | FileCheck %s --check-prefix=FCGLMD
+// RUN: %if dxil-1-8  \
+// RUN:   %{          \
+// RUN: %dxc -T lib_6_8 %s | FileCheck %s --check-prefixes=MD \
+// RUN:   %}
+
+// RUN: %if dxil-1-8  \
+// RUN:   %{          \
+// RUN: %dxc -T lib_6_8 -Od %s | FileCheck %s --check-prefixes=MD \
+// RUN:   %}
+
+// RUN: %if dxil-1-8  \
+// RUN:   %{          \
+// RUN: %dxc -T lib_6_8 -Zi %s | FileCheck %s --check-prefixes=MD \
+// RUN:   %}
+
+// RUN: %if dxil-1-8  \
+// RUN:   %{          \
+// RUN: %dxc -T lib_6_8 -fcgl %s | FileCheck %s --check-prefix=FCGLMD \
+// RUN:   %}
 
 // Verify correct metadata annotations for different node shader input and outputs.
 

tools/clang/test/lit.cfg

@@ -503,6 +503,33 @@ if config.enable_backtrace == "1":
 if config.spirv:
     config.available_features.add("spirv")
 
+# Check supported dxil version
+def get_dxil_version():
+    result = subprocess.run([lit.util.which('dxc', llvm_tools_dir), "--version"], stdout=subprocess.PIPE)
+    output = result.stdout.decode("utf-8")
+    pat = re.compile(r"(dxcompiler.dll|libdxcompiler.so|libdxcompiler.dylib): (?P<dxcMajor>[0-9]+)\.(?P<dxcMinor>[0-9]+).*(; (dxil.dll|libdxil.so): (?P<dxilMajor>[0-9]+)\.(?P<dxilMinor>[0-9]+))?")
+    m = pat.search(output)
+    dxcMajor = int(m.group("dxcMajor"))
+    dxcMinor = int(m.group("dxcMinor"))
+
+    if None == m.group("dxilMajor"):
+        return dxcMajor, dxcMinor
+    dxilMajor = int(m.group("dxilMajor"))
+    dxilMinor = int(m.group("dxilMinor"))
+    if dxcMajor < dxilMajor:
+        return dxcMajor, dxcMinor
+    if dxcMajor > dxilMajor:
+        return dxilMajor, dxilMinor
+    if dxcMinor < dxilMinor:
+        return dxcMajor, dxcMinor
+    return dxilMajor, dxilMinor
+
+dxilMajor, dxilMinor = get_dxil_version()
+if dxilMajor == 1:
+    for i in range(0, dxilMinor + 1):
+        config.available_features.add(f"dxil-1-{i}")
+
+
 # Check if we should run long running tests.
 if lit_config.params.get("run_long_tests", None) == "true":
     config.available_features.add("long_tests")