提交 d62c7778 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3215 from nouiz/cnmem

Integrate the latest version of CNMEM
...@@ -381,6 +381,8 @@ public: ...@@ -381,6 +381,8 @@ public:
mStream = stream; mStream = stream;
#ifdef CUDA_API_PER_THREAD_DEFAULT_STREAM #ifdef CUDA_API_PER_THREAD_DEFAULT_STREAM
mIsStreamBlocking = false; mIsStreamBlocking = false;
#elif CUDART_VERSION < 5050
mIsStreamBlocking = true;
#else #else
unsigned flags = 0; unsigned flags = 0;
CNMEM_CHECK_CUDA(cudaStreamGetFlags(mStream, &flags)); CNMEM_CHECK_CUDA(cudaStreamGetFlags(mStream, &flags));
......
...@@ -253,7 +253,7 @@ cnmemStatus_t CNMEM_API cnmemPrintMemoryState(FILE *file, cudaStream_t stream); ...@@ -253,7 +253,7 @@ cnmemStatus_t CNMEM_API cnmemPrintMemoryState(FILE *file, cudaStream_t stream);
/** /**
* \brief Converts a cnmemStatus_t value to a string. * \brief Converts a cnmemStatus_t value to a string.
*/ */
const char* CNMEM_API cnmemGetErrorString(cnmemStatus_t status); const char CNMEM_API * cnmemGetErrorString(cnmemStatus_t status);
/* ********************************************************************************************* */ /* ********************************************************************************************* */
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论