Hi Subramanya,
Thanks for the quick reply! So, the cell batch id does change between different dof_handlers. I think that's a caveat worth mentioning in the tutorial.
Also, I'm curious about the partition algorithm, what is actually related to the cell partition, the Mesh, the FE, the partition_scheme (none/color/parallel etc.)? I didn't find the explicit initialization in `matrix_free.templates.h`. It should within the function `compute_dof_info`.
Regards,
Chengjiang Yin