Add test case for GPU