소스 검색

Restore unregister spuriously removed

Samuel Thibault 13 년 전
부모
커밋
3265678122
1개의 변경된 파일4개의 추가작업 그리고 0개의 파일을 삭제
  1. 4 0
      tests/datawizard/gpu_register.c

+ 4 - 0
tests/datawizard/gpu_register.c

@@ -95,6 +95,10 @@ int main(int argc, char **argv)
 	ret = starpu_task_wait_for_all();
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_wait_for_all");
 
+	starpu_data_unpartition(handle, 0);
+	starpu_data_unregister(handle);
+
+	cudaSetDevice(devid);
 	cures = cudaMemcpy(foo, foo_gpu, size * sizeof(*foo_gpu), cudaMemcpyDeviceToHost);
 	if (STARPU_UNLIKELY(cures))
 		STARPU_CUDA_REPORT_ERROR(cures);