refactor(dfir_lang): remove stratum, add push codegen for lattice_fold, lattice_reduce, _lattice_fold_batch

PR: #2970

Author: MingweiSamuel

dfir_lang/src/graph/ops/_lattice_fold_batch.rs

@@ -2,7 +2,7 @@ use quote::{quote_spanned, ToTokens};
 use syn::parse_quote;
 
 use super::{
-    DelayType, OpInstGenerics, OperatorCategory, OperatorConstraints, OperatorInstance,
+    OpInstGenerics, OperatorCategory, OperatorConstraints, OperatorInstance,
     OperatorWriteOutput, PortListSpec, WriteContextArgs, RANGE_0, RANGE_1,
 };
 
@@ -44,7 +44,7 @@ pub const _LATTICE_FOLD_BATCH: OperatorConstraints = OperatorConstraints {
     flo_type: None,
     ports_inn: Some(|| PortListSpec::Fixed(parse_quote! { input, signal })),
     ports_out: None,
-    input_delaytype_fn: |_| Some(DelayType::MonotoneAccum),
+    input_delaytype_fn: |_| None,
     write_fn: |wc @ &WriteContextArgs {
                    ident,
                    op_span,

dfir_lang/src/graph/ops/lattice_fold.rs

@@ -1,7 +1,7 @@
 use syn::parse_quote_spanned;
 
 use super::{
-    DelayType, OperatorCategory, OperatorConstraints, WriteContextArgs,
+    OperatorCategory, OperatorConstraints, WriteContextArgs,
     RANGE_0, RANGE_1,
 };
 
@@ -41,17 +41,14 @@ pub const LATTICE_FOLD: OperatorConstraints = OperatorConstraints {
     flo_type: None,
     ports_inn: None,
     ports_out: None,
-    input_delaytype_fn: |_| Some(DelayType::MonotoneAccum),
+    input_delaytype_fn: |_| None,
     write_fn: |wc @ &WriteContextArgs {
                    root,
-                   is_pull,
                    op_span,
                    arguments,
                    ..
                },
                diagnostics| {
-        assert!(is_pull);
-
         let first_arg = &arguments[0];
 
         let arguments = &parse_quote_spanned! {op_span=>

dfir_lang/src/graph/ops/lattice_reduce.rs

@@ -2,7 +2,7 @@ use quote::quote_spanned;
 use syn::parse_quote_spanned;
 
 use super::{
-    DelayType, OperatorCategory, OperatorConstraints, OperatorWriteOutput, RANGE_0, RANGE_1,
+    OperatorCategory, OperatorConstraints, OperatorWriteOutput, RANGE_0, RANGE_1,
     WriteContextArgs,
 };
 
@@ -42,7 +42,7 @@ pub const LATTICE_REDUCE: OperatorConstraints = OperatorConstraints {
     flo_type: None,
     ports_inn: None,
     ports_out: None,
-    input_delaytype_fn: |_| Some(DelayType::MonotoneAccum),
+    input_delaytype_fn: |_| None,
     write_fn: |wc @ &WriteContextArgs {
                    root,
                    inputs,
@@ -51,8 +51,6 @@ pub const LATTICE_REDUCE: OperatorConstraints = OperatorConstraints {
                    ..
                },
                diagnostics| {
-        assert!(is_pull);
-
         let arguments = &parse_quote_spanned! {op_span=>
             |acc, item| { #root::lattices::Merge::<_>::merge(acc, item); }
         };
@@ -68,24 +66,27 @@ pub const LATTICE_REDUCE: OperatorConstraints = OperatorConstraints {
             write_tick_end,
         } = (super::reduce::REDUCE.write_fn)(&wc, diagnostics)?;
 
-        assert_eq!(1, inputs.len());
-        let input = &inputs[0];
-
-        let write_iterator = quote_spanned! {op_span=>
-            let #input = {
-                #[inline(always)]
-                fn check_inputs<Lat, Prev>(
-                    input: Prev,
-                ) -> impl #root::dfir_pipes::pull::Pull<Item = Lat, Meta = Prev::Meta, CanPend = Prev::CanPend, CanEnd = Prev::CanEnd>
-                where
-                    Lat: #root::lattices::Merge<Lat>,
-                    Prev: #root::dfir_pipes::pull::Pull<Item = Lat>,
-                {
-                    input
-                }
-                check_inputs(#input)
-            };
-            #write_iterator
+        let write_iterator = if is_pull {
+            assert_eq!(1, inputs.len());
+            let input = &inputs[0];
+            quote_spanned! {op_span=>
+                let #input = {
+                    #[inline(always)]
+                    fn check_inputs<Lat, Prev>(
+                        input: Prev,
+                    ) -> impl #root::dfir_pipes::pull::Pull<Item = Lat, Meta = Prev::Meta, CanPend = Prev::CanPend, CanEnd = Prev::CanEnd>
+                    where
+                        Lat: #root::lattices::Merge<Lat>,
+                        Prev: #root::dfir_pipes::pull::Pull<Item = Lat>,
+                    {
+                        input
+                    }
+                    check_inputs(#input)
+                };
+                #write_iterator
+            }
+        } else {
+            write_iterator
         };
         Ok(OperatorWriteOutput {
             write_prologue,