change head 1d implementation to have some unrolled steps