Split the unittest to Gpu and Cpu.