提交 6aa85f9e authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Merge pull request #2009 from abergeron/fix_cublas_init

Fix compile on cuda < 5.0.
......@@ -3587,9 +3587,11 @@ cublas_init()
cublasSetStream(handle, NULL);
// Pointer to scalars are on the host (also default)
cublasSetPointerMode(handle, CUBLAS_POINTER_MODE_HOST);
#if CUDA_VERSION >= 5000
// atomics can be used in kernels to speed up operations (not default)
// This may lead to a slight variance from run to run in some operations
cublasSetAtomicsMode(handle, CUBLAS_ATOMICS_ALLOWED);
#endif
return 0;
}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论