Fix compilation crash on GPU.

We tried to use the CPU code for the GPU op.