Fix compilation crash on GPU.

We tried to use the CPU code for the GPU op.
Merge pull request #1 from nouiz/abalkin-take-op-c-code-clean

Fix compilation crash on GPU.