Should be done similar to how we handle it for sampling after https://github.com/stan-dev/cmdstanr/pull/185 is merged.