I know that K-1 partners that are working partners (earned income) can participate in 401ks. But, I am not sure on how the money flows into the 401k. For accounting purposes, do those partners take distributions from the company to put towards their 401k deferrals at the same time that they do payroll for their W-2 employees? Or how does it work?

Thank you all for your help!
Jennifer