Commit 55fd4de0 by liyinqiao

Merge with ZHANG Yuhao branch

Fix the bugs in CUDA 10.
parent 857114fe
...@@ -147,7 +147,11 @@ void XStream::StreamSynchronize() ...@@ -147,7 +147,11 @@ void XStream::StreamSynchronize()
void XStream::ThreadSynchronize() void XStream::ThreadSynchronize()
{ {
#ifdef USE_CUDA #ifdef USE_CUDA
#if CUDART_VERSION < 10000
cudaThreadSynchronize(); cudaThreadSynchronize();
#else
ShowNTErrors("TODO!");
#endif
#endif #endif
} }
......
...@@ -834,7 +834,7 @@ int CompXFloat(const void * a, const void * b) ...@@ -834,7 +834,7 @@ int CompXFloat(const void * a, const void * b)
void ResetGPUDevices() void ResetGPUDevices()
{ {
#ifdef USE_CUDA #ifdef USE_CUDA
#if CUDART_VERSION < 10000
cudaThreadExit(); cudaThreadExit();
return; return;
...@@ -845,6 +845,9 @@ void ResetGPUDevices() ...@@ -845,6 +845,9 @@ void ResetGPUDevices()
cudaSetDevice(i); cudaSetDevice(i);
cudaDeviceReset(); cudaDeviceReset();
}*/ }*/
#elif
ShowNTErrors("TODO!");
#endif
#endif #endif
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论