Cache fiat element on ufl element.

This trivial change causes fiat_from_ufl_element to drop from 5s to
nothing on the wave demo.