[IMPROVE] update energy.cu

it is still a draft