that's it?

private-attribution · Dec 12, 2024 · 982dc32 · 982dc32
1 parent ff37b8a
commit 982dc32
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/ipa-core/src/query/runner/hybrid.rs b/ipa-core/src/query/runner/hybrid.rs
@@ -46,6 +46,7 @@ use crate::{
         replicated::semi_honest::AdditiveShare as Replicated, BitDecomposed, TransposeFrom,
         Vectorizable,
     },
+    seq_join::seq_join,
     sharding::{ShardConfiguration, Sharded},
 };
 
@@ -130,10 +131,12 @@ where
                 }))
             })
             .try_flatten()
-            .take(sz);
+            .take(sz)
+            .map(|v| async move { v });
+
         let (decrypted_reports, resharded_tags) = reshard_aad(
             ctx.narrow(&HybridStep::ReshardByTag),
-            stream,
+            seq_join(ctx.active_work(), stream),
             |ctx, _, tag| tag.shard_picker(ctx.shard_count()),
         )
         .await?;