diff --git a/tests/analytical_visualization/app.py b/tests/analytical_visualization/app.py index 0285b02..0c1da5d 100644 --- a/tests/analytical_visualization/app.py +++ b/tests/analytical_visualization/app.py @@ -185,6 +185,18 @@ with st.sidebar.expander("Parallelism", expanded=True): f"{_default_suggestion.pes_used} PEs " f"(cp_placement={_default_suggestion.cp_placement})" ) + # Memory-min apply — snap sliders back to what the caption says. + if st.button("Apply memory-min", width='stretch', + key="_sb_apply_memmin", + help="Set sliders to the memory-min config shown above " + "(smallest deployment that fits)."): + st.session_state["cp"] = _snap(_default_suggestion.cp, _CP_OPTS) + st.session_state["tp"] = _snap(_default_suggestion.tp, _TP_OPTS) + st.session_state["pp"] = _snap(_default_suggestion.pp, _PP_OPTS) + st.session_state["dp"] = 1 + st.session_state["cp_placement"] = _default_suggestion.cp_placement + st.session_state["_pl_active_scope"] = "full" + st.rerun() # Three latency-optimal apply buttons — pick a scope, load the # latency-min Pareto config into the sidebar sliders. from tests.analytical_visualization.auto_explore import (