subgroupballot metal out

gfx-rs · Sep 30, 2023 · 7ad4bd5 · 7ad4bd5
1 parent b851a52
commit 7ad4bd5
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/back/msl/writer.rs b/src/back/msl/writer.rs
@@ -1905,6 +1905,7 @@ impl<W: Write> Writer<W> {
             crate::Expression::CallResult(_)
             | crate::Expression::AtomicResult { .. }
             | crate::Expression::WorkGroupUniformLoadResult { .. }
+            | crate::Expression::SubgroupBallotResult
             | crate::Expression::RayQueryProceedResult => {
                 unreachable!()
             }
@@ -1963,7 +1964,6 @@ impl<W: Write> Writer<W> {
                 }
                 write!(self.out, "}}")?;
             }
-            crate::Expression::SubgroupBallotResult => todo!(),
         }
         Ok(())
     }
@@ -2977,7 +2977,13 @@ impl<W: Write> Writer<W> {
                         }
                     }
                 }
-                crate::Statement::SubgroupBallot { result } => todo!(),
+                crate::Statement::SubgroupBallot { result } => {
+                    write!(self.out, "{level}")?;
+                    let name = self.namer.call("");
+                    self.start_baking_expression(result, &context.expression, &name)?;
+                    self.named_expressions.insert(result, name);
+                    write!(self.out, "{NAMESPACE}::simd_active_threads_mask();")?;
+                }
             }
         }