Function make_context

cake-core/tests/unit_tests/test_blocks.rs:24–42

(cfg: Config, vb: VarBuilder<'static>)

// ---------------------------------------------------------------------------

fn make_context(cfg: Config, vb: VarBuilder<'static>) -> Context {
    let cache = Cache::new(true, DType::F32, &cfg, &Device::Cpu).unwrap();
    Context {
        args: Args::default(),
        dtype: DType::F32,
        topology: Topology::new(),
        data_path: PathBuf::from("."),
        device: Device::Cpu,
        config: Some(cfg),
        cache: Some(cache),
        var_builder: Some(vb),
        text_model_arch: TextModelArch::Auto,
        quant: Arc::new(cake_core::utils::NoQuantization),
        listener_override: Arc::new(Mutex::new(None)),
            tensor_storage: None,
            layer_devices: None,
        backend: Arc::new(cake_core::backends::CpuBackend::new()),
    }
}

// ---------------------------------------------------------------------------
// Helper: build VarBuilder for a standard block (attn + mlp + norms)