Firedrake forms: only register assembly timing