Run on CPU with the same code