Add primitive set of arch instructions (#520)

Author: XAMPPRocky

crates/spirv-std/src/arch.rs

@@ -9,10 +9,12 @@ use crate::{scalar::Scalar, vector::Vector};
 #[cfg(feature = "const-generics")]
 mod arithmetic;
 mod derivative;
+mod primitive;
 
 #[cfg(feature = "const-generics")]
 pub use arithmetic::*;
 pub use derivative::*;
+pub use primitive::*;
 
 /// Result is true if any component of `vector` is true, otherwise result is
 /// false.

crates/spirv-std/src/arch/primitive.rs

@@ -0,0 +1,72 @@
+/// Emits the current values of all output variables to the current output
+/// primitive. After execution, the values of all output variables
+/// are undefined.  Requires capability `Geometry`.
+///
+/// # Safety
+/// This instruction must only be used when only one stream is present.
+#[spirv_std_macros::gpu_only]
+#[doc(alias = "OpEmitVertex")]
+#[inline]
+pub unsafe fn emit_vertex() {
+    asm! {
+        "OpEmitVertex",
+    }
+}
+
+/// Finish the current primitive and start a new one. No vertex is emitted.
+/// Requires capability `Geometry`.
+///
+/// # Safety
+/// This instruction must only be used when only one stream is present.
+#[spirv_std_macros::gpu_only]
+#[doc(alias = "OpEndPrimitive")]
+#[inline]
+pub unsafe fn end_primitive() {
+    asm! {
+        "OpEndPrimitive",
+    }
+}
+
+/// Emits the current values of all output variables to the current output
+/// primitive. After execution, the values of all output variables
+/// are undefined.
+///
+/// `STREAM` is the output-primitive stream number.
+///
+/// Requires capability `GeometryStreams`.
+///
+/// # Safety
+/// This instruction must only be used when multiple streams are present.
+#[spirv_std_macros::gpu_only]
+#[doc(alias = "OpEmitStreamVertex")]
+#[inline]
+#[cfg(feature = "const-generics")]
+pub unsafe fn emit_stream_vertex<const STREAM: i64>() {
+    asm! {
+        "%i64 = OpTypeInt 64 1",
+        "%stream = OpConstant %i64 {stream}",
+        "OpEmitStreamVertex %stream",
+        stream = const STREAM,
+    }
+}
+
+/// Finish the current primitive and start a new one. No vertex is emitted.
+///
+/// `STREAM` is the output-primitive stream number.
+///
+/// Requires capability `GeometryStreams`.
+///
+/// # Safety
+/// This instruction must only be used when multiple streams are present.
+#[spirv_std_macros::gpu_only]
+#[doc(alias = "OpEndStreamPrimitive")]
+#[inline]
+#[cfg(feature = "const-generics")]
+pub unsafe fn end_stream_primitive<const STREAM: i64>() {
+    asm! {
+        "%i64 = OpTypeInt 64 1",
+        "%stream = OpConstant %i64 {stream}",
+        "OpEndStreamPrimitive %stream",
+        stream = const STREAM,
+    }
+}

tests/ui/arch/emit_stream_vertex.rs

@@ -0,0 +1,9 @@
+// build-pass
+
+#[spirv(geometry(input_lines = 2, output_points = 2))]
+pub fn main() {
+    unsafe {
+        asm!("OpCapability GeometryStreams");
+        spirv_std::arch::emit_stream_vertex::<2>();
+    };
+}

tests/ui/arch/emit_vertex.rs

@@ -0,0 +1,9 @@
+// build-pass
+
+#[spirv(geometry(input_lines = 2, output_points = 2))]
+pub fn main() {
+    unsafe {
+        asm!("OpCapability Geometry");
+        spirv_std::arch::emit_vertex();
+    };
+}

tests/ui/arch/end_primitive.rs

@@ -0,0 +1,9 @@
+// build-pass
+
+#[spirv(geometry(input_lines = 2, output_points = 2))]
+pub fn main() {
+    unsafe {
+        asm!("OpCapability Geometry");
+        spirv_std::arch::end_primitive();
+    };
+}

tests/ui/arch/end_stream_primitive.rs

@@ -0,0 +1,9 @@
+// build-pass
+
+#[spirv(geometry(input_lines = 2, output_points = 2))]
+pub fn main() {
+    unsafe {
+        asm!("OpCapability GeometryStreams");
+        spirv_std::arch::end_stream_primitive::<2>();
+    };
+}