Add the code CPU/GPU