compute heat flux in lammps