Add a Minimal Mesh Shader Test that runs on the DX12 backend. (#1175)

Author: manon-traverse

include/API/Encoder.h

@@ -109,6 +109,10 @@ class RenderEncoder : public CommandEncoder {
                                     uint32_t InstanceCount,
                                     uint32_t FirstVertex = 0,
                                     uint32_t FirstInstance = 0) = 0;
+
+  virtual llvm::Error dispatchMesh(const PipelineState &PSO,
+                                   uint32_t GroupCountX, uint32_t GroupCountY,
+                                   uint32_t GroupCountZ) = 0;
 };
 
 } // namespace offloadtest

include/Support/Pipeline.h

@@ -26,15 +26,25 @@
 
 namespace offloadtest {
 
-enum class Stages { Compute, Vertex, Pixel };
+enum class Stages {
+  // Compute
+  Compute,
+
+  // Traditional Raster
+  Vertex,
+  Pixel,
+
+  // Mesh Shader Raster
+  Amplification,
+  Mesh
+};
 inline constexpr std::array AllStages = {
-    Stages::Compute,
-    Stages::Vertex,
-    Stages::Pixel,
+    Stages::Compute,       Stages::Vertex, Stages::Pixel,
+    Stages::Amplification, Stages::Mesh,
 };
 inline constexpr size_t NumStages = AllStages.size();
 
-enum class ShaderPipelineKind { Compute, TraditionalRaster };
+enum class ShaderPipelineKind { Compute, TraditionalRaster, MeshShaderRaster };
 
 enum class Rule { BufferExact, BufferFloatULP, BufferFloatEpsilon };
 
@@ -503,6 +513,12 @@ struct Pipeline {
   bool isTraditionalRaster() const {
     return Kind == ShaderPipelineKind::TraditionalRaster;
   }
+  bool isMeshShaderRaster() const {
+    return Kind == ShaderPipelineKind::MeshShaderRaster;
+  }
+  bool isRaster() const {
+    return isTraditionalRaster() || isMeshShaderRaster();
+  }
 };
 } // namespace offloadtest
 
@@ -713,6 +729,8 @@ template <> struct ScalarEnumerationTraits<offloadtest::Stages> {
     ENUM_CASE(Compute);
     ENUM_CASE(Vertex);
     ENUM_CASE(Pixel);
+    ENUM_CASE(Amplification);
+    ENUM_CASE(Mesh);
 #undef ENUM_CASE
   }
 };

lib/API/DX/DXFeatures.cpp

@@ -34,6 +34,15 @@ static ArrayRef<EnumEntry<RootSignature>> getRootSignatures() {
   return ArrayRef(RootSignatureNames);
 }
 
+#define MESH_SHADER_TIER_ENUM(NewCase, Str, Value) {#Str, NewCase},
+static const EnumEntry<directx::MeshShaderTier> MeshShaderTierNames[]{
+#include "DXFeatures.def"
+};
+
+static ArrayRef<EnumEntry<MeshShaderTier>> getMeshShaderTiers() {
+  return ArrayRef(MeshShaderTierNames);
+}
+
 template <typename T>
 static std::string enumEntryToString(ArrayRef<EnumEntry<T>> EnumValues,
                                      T Value) {
@@ -52,3 +61,8 @@ std::string CapabilityPrinter<directx::RootSignature>::toString(
     const directx::RootSignature &V) {
   return enumEntryToString(getRootSignatures(), V);
 }
+
+std::string CapabilityPrinter<directx::MeshShaderTier>::toString(
+    const directx::MeshShaderTier &V) {
+  return enumEntryToString(getMeshShaderTiers(), V);
+}

lib/API/DX/DXFeatures.def

@@ -22,9 +22,16 @@ ROOT_SIGNATURE_ENUM(RootSig_v1_2, 1.2, 0x3)
 #undef ROOT_SIGNATURE_ENUM
 #endif
 
+#ifdef MESH_SHADER_TIER_ENUM
+MESH_SHADER_TIER_ENUM(MeshShaderNotSupported, NotSupported, 0)
+MESH_SHADER_TIER_ENUM(MeshShaderTier1, Tier1,10)
+#undef MESH_SHADER_TIER_ENUM
+#endif
+
 #ifdef D3D_FEATURE_ENUM
 D3D_FEATURE_ENUM(directx::ShaderModel, HighestShaderModel)
 D3D_FEATURE_ENUM(directx::RootSignature, HighestRootSignatureVersion)
+D3D_FEATURE_ENUM(directx::MeshShaderTier, MeshShaderTier)
 #undef D3D_FEATURE_ENUM
 #endif
 

lib/API/DX/DXFeatures.h

@@ -32,6 +32,11 @@ enum RootSignature {
 #include "DXFeatures.def"
 };
 
+#define MESH_SHADER_TIER_ENUM(NewCase, Str, Value) NewCase = Value,
+enum MeshShaderTier {
+#include "DXFeatures.def"
+};
+
 } // namespace directx
 
 template <> struct CapabilityPrinter<directx::ShaderModel> {
@@ -42,6 +47,10 @@ template <> struct CapabilityPrinter<directx::RootSignature> {
   static std::string toString(const directx::RootSignature &V);
 };
 
+template <> struct CapabilityPrinter<directx::MeshShaderTier> {
+  static std::string toString(const directx::MeshShaderTier &V);
+};
+
 } // namespace offloadtest
 
 #endif // OFFLOADTEST_API_DXFEATURES_H