Переглянути джерело

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