Merge pull request #204 from astnmsn/fix/wasm-async-recursion

set async_recursion to ?Send for wasm32 arch
webonnx · Mar 19, 2024 · df34358 · df34358
2 parents 7880ed8 + af090b5
commit df34358
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/wonnx/src/optimizer.rs b/wonnx/src/optimizer.rs
@@ -292,7 +292,8 @@ impl<'model> Optimizer<'model> {
     }
 
     /// Optimize a branch of a graph (memoized)
-    #[async_recursion]
+    #[cfg_attr(target_arch = "wasm32", async_recursion(?Send))]
+    #[cfg_attr(not(target_arch = "wasm32"), async_recursion)]
     pub async fn optimize(
         &mut self,
         node: Arc<Node<'model>>,
@@ -310,7 +311,8 @@ impl<'model> Optimizer<'model> {
 
     /// Optimize a branch of a graph. Takes a node an attempts to form a chain of nodes with single (dynamic) inputs by
     /// traversing towards the inputs.
-    #[async_recursion]
+    #[cfg_attr(target_arch = "wasm32", async_recursion(?Send))]
+    #[cfg_attr(not(target_arch = "wasm32"), async_recursion)]
     async fn optimize_actual(
         &mut self,
         node: Arc<Node<'model>>,