mesh shader ext: compiletest for mesh shaders, from @BeastLe9enD

Author: BeastLe9enD

tests/ui/arch/mesh_shader_output_lines.rs

@@ -0,0 +1,27 @@
+// build-pass
+// only-vulkan1.2
+// compile-flags: -Ctarget-feature=+MeshShadingEXT,+ext:SPV_EXT_mesh_shader
+
+use spirv_std::arch::set_mesh_outputs_ext;
+use spirv_std::glam::{UVec2, Vec4};
+use spirv_std::spirv;
+
+#[spirv(mesh_ext(
+    threads(1),
+    output_vertices = 2,
+    output_primitives_ext = 1,
+    output_lines_ext
+))]
+pub fn main(
+    #[spirv(position)] positions: &mut [Vec4; 2],
+    #[spirv(primitive_line_indices_ext)] indices: &mut [UVec2; 1],
+) {
+    unsafe {
+        set_mesh_outputs_ext(2, 1);
+    }
+
+    positions[0] = Vec4::new(-0.5, 0.5, 0.0, 1.0);
+    positions[1] = Vec4::new(0.5, 0.5, 0.0, 1.0);
+
+    indices[0] = UVec2::new(0, 1);
+}

tests/ui/arch/mesh_shader_output_points.rs

@@ -0,0 +1,26 @@
+// build-pass
+// only-vulkan1.2
+// compile-flags: -Ctarget-feature=+MeshShadingEXT,+ext:SPV_EXT_mesh_shader
+
+use spirv_std::arch::set_mesh_outputs_ext;
+use spirv_std::glam::{UVec2, Vec4};
+use spirv_std::spirv;
+
+#[spirv(mesh_ext(
+    threads(1),
+    output_vertices = 1,
+    output_primitives_ext = 1,
+    output_points
+))]
+pub fn main(
+    #[spirv(position)] positions: &mut [Vec4; 1],
+    #[spirv(primitive_point_indices_ext)] indices: &mut [u32; 1],
+) {
+    unsafe {
+        set_mesh_outputs_ext(1, 1);
+    }
+
+    positions[0] = Vec4::new(-0.5, 0.5, 0.0, 1.0);
+
+    indices[0] = 0;
+}

tests/ui/arch/mesh_shader_output_triangles.rs

@@ -0,0 +1,28 @@
+// build-pass
+// only-vulkan1.2
+// compile-flags: -Ctarget-feature=+MeshShadingEXT,+ext:SPV_EXT_mesh_shader
+
+use spirv_std::arch::set_mesh_outputs_ext;
+use spirv_std::glam::{UVec3, Vec4};
+use spirv_std::spirv;
+
+#[spirv(mesh_ext(
+    threads(1),
+    output_vertices = 3,
+    output_primitives_ext = 1,
+    output_triangles_ext
+))]
+pub fn main(
+    #[spirv(position)] positions: &mut [Vec4; 3],
+    #[spirv(primitive_triangle_indices_ext)] indices: &mut [UVec3; 1],
+) {
+    unsafe {
+        set_mesh_outputs_ext(3, 1);
+    }
+
+    positions[0] = Vec4::new(-0.5, 0.5, 0.0, 1.0);
+    positions[1] = Vec4::new(0.5, 0.5, 0.0, 1.0);
+    positions[2] = Vec4::new(0.0, -0.5, 0.0, 1.0);
+
+    indices[0] = UVec3::new(0, 1, 2);
+}

tests/ui/arch/mesh_shader_payload.rs

@@ -0,0 +1,35 @@
+// build-pass
+// only-vulkan1.2
+// compile-flags: -Ctarget-feature=+MeshShadingEXT,+ext:SPV_EXT_mesh_shader
+
+use spirv_std::arch::set_mesh_outputs_ext;
+use spirv_std::glam::{UVec3, Vec4};
+use spirv_std::spirv;
+
+pub struct Payload {
+    pub first: f32,
+    pub second: f32,
+    pub third: f32
+}
+
+#[spirv(mesh_ext(
+    threads(1),
+    output_vertices = 3,
+    output_primitives_ext = 1,
+    output_triangles_ext
+))]
+pub fn main(
+    #[spirv(position)] positions: &mut [Vec4; 3],
+    #[spirv(primitive_triangle_indices_ext)] indices: &mut [UVec3; 1],
+    #[spirv(task_payload_workgroup_ext)] payload: &Payload
+) {
+    unsafe {
+        set_mesh_outputs_ext(3, 1);
+    }
+
+    positions[0] = payload.first * Vec4::new(-0.5, 0.5, 0.0, 1.0);
+    positions[1] = payload.second * Vec4::new(0.5, 0.5, 0.0, 1.0);
+    positions[2] = payload.third * Vec4::new(0.0, -0.5, 0.0, 1.0);
+
+    indices[0] = UVec3::new(0, 1, 2);
+}

tests/ui/arch/task_shader.rs

@@ -0,0 +1,13 @@
+// build-pass
+// only-vulkan1.2
+// compile-flags: -Ctarget-feature=+MeshShadingEXT,+ext:SPV_EXT_mesh_shader
+
+use spirv_std::spirv;
+use spirv_std::arch::emit_mesh_tasks_ext;
+
+#[spirv(task_ext(threads(1)))]
+pub fn main() {
+    unsafe {
+        emit_mesh_tasks_ext(1, 2, 3);
+    }
+}

tests/ui/arch/task_shader_payload.rs

@@ -0,0 +1,21 @@
+// build-pass
+// only-vulkan1.2
+// compile-flags: -Ctarget-feature=+MeshShadingEXT,+ext:SPV_EXT_mesh_shader
+
+use spirv_std::spirv;
+use spirv_std::arch::emit_mesh_tasks_ext;
+
+pub struct Payload {
+    pub first: u32,
+    pub second: i32
+}
+
+#[spirv(task_ext(threads(1)))]
+pub fn main(#[spirv(task_payload_workgroup_ext)] payload: &mut Payload) {
+    payload.first = 1;
+    payload.second = 2;
+
+    unsafe {
+        emit_mesh_tasks_ext(3, 4, 5);
+    }
+}