rlx-flow 0.2.1

Block assembly-line API for RLX model builders — fusion-first, config-driven
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

// RLX — versatile ML compiler + runtime.
// Copyright (C) 2026 Eugene Hauptmann, Nataliya Kosmyna.

//! Flow-level extension wiring — applies [`rlx_ir::hir_extension`] after build.

use rlx_ir::hir::HirModule;
use rlx_ir::{apply_hir_extensions, apply_hir_extensions_named};

/// Names of HIR extensions to apply when building this flow (empty = all registered).
#[derive(Debug, Clone, Default, PartialEq, Eq)]
pub struct FlowExtensionPlan {
    pub enabled: Vec<String>,
    pub apply_all: bool,
}

impl FlowExtensionPlan {
    pub fn all() -> Self {
        Self {
            enabled: Vec::new(),
            apply_all: true,
        }
    }

    pub fn only(names: impl IntoIterator<Item = impl Into<String>>) -> Self {
        Self {
            enabled: names.into_iter().map(Into::into).collect(),
            apply_all: false,
        }
    }

    pub fn apply(&self, hir: &mut HirModule) {
        if self.apply_all {
            apply_hir_extensions(hir);
        } else {
            let refs: Vec<&str> = self.enabled.iter().map(String::as_str).collect();
            apply_hir_extensions_named(hir, &refs);
        }
    }
}