a2xx/a3xx: more registers for kernel