Wiring up offload_opt_states (#8186)