Add kinetic_energy_enstrophy to test_gpu.