Function build_trace_with_budget 

pub fn build_trace_with_budget(
    witness: VmWitness,
    max_prover_memory_bytes: u64,
) -> Result<VmTrace, ExecutionError>

Same as build_trace, but with an explicit memory budget instead of the default.