def test_accumulate_grads_gpu_to_gpu(self): device_id = cuda.Device().id self.setup_gpu(device_id) self.check_accumulate_grads_from_gpu(device_id)