gpu: Actually fix PE to GPU mapping