|
@@ -290,7 +290,11 @@ static void measure_bandwidth_between_host_and_dev_on_numa_with_cuda(int dev, in
|
|
|
|
|
|
cudaFree(d_buffer);
|
|
|
|
|
|
+#if CUDART_VERSION >= 4000
|
|
|
+ cudaDeviceReset();
|
|
|
+#else
|
|
|
cudaThreadExit();
|
|
|
+#endif
|
|
|
}
|
|
|
|
|
|
#ifdef STARPU_HAVE_CUDA_MEMCPY_PEER
|
|
@@ -400,7 +404,11 @@ static void measure_bandwidth_between_dev_and_dev_cuda(int src, int dst)
|
|
|
cudaSetDevice(src);
|
|
|
cudaFree(s_buffer);
|
|
|
|
|
|
+#if CUDART_VERSION >= 4000
|
|
|
+ cudaDeviceReset();
|
|
|
+#else
|
|
|
cudaThreadExit();
|
|
|
+#endif
|
|
|
}
|
|
|
#endif
|
|
|
#endif
|