pt: remove env.DEVICE in all forward functions#3330

njzjz merged 3 commits intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom

njzjz:pt-remove-env-devicenjzjz/deepmd-kit:pt-remove-env-deviceCopy head branch name to clipboard