fixed the multiple gpu revert issue