瀏覽代碼

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);