Revert "fix GPU"

This reverts commit 8122b8b1ff3ce50a7eaefde1e30488a1ec3bb447.