Fix size of equation_of_state and isoneutral_mixing, add tke (#651)
Summary:
Use the datapoint size=1048576 in https://github.com/dionhaefner/pyhpc-benchmarks/blob/master/README.md
Profiling results are below.
pyhpc_equation_of_state eval:

pyhpc_isoneutral_mixing eval:

pyhpc_turbulent_kinetic_energy eval:

Pull Request resolved: https://github.com/pytorch/benchmark/pull/651
Reviewed By: aaronenyeshi
Differential Revision: D33163155
Pulled By: xuzhao9
fbshipit-source-id: d65447272aacb04ac126487b256ab8c789af32d0