def test_accumulate_grads_gpu_to_cpu(self): self.setup_cpu() self.check_accumulate_grads_from_gpu(cuda.Device().id)