gpu: store grid size in ppcg_kernel