未验证 提交 85c77609 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #8716 from luotao1/cuda_version

fix only shared variables could be declared as static in the device code
...@@ -158,7 +158,13 @@ HOSTDEVICE int64_t& indexer<0>(Dim<0>& dim, int idx) { ...@@ -158,7 +158,13 @@ HOSTDEVICE int64_t& indexer<0>(Dim<0>& dim, int idx) {
#else #else
PADDLE_ASSERT(false); PADDLE_ASSERT(false);
#endif #endif
#if (defined __CUDA_ARCH__) && (CUDA_VERSION < 8000)
// On CUDA versions previous to 8.0, only __shared__ variables
// could be declared as static in the device code.
int64_t head = 0;
#else
static int64_t head = 0; static int64_t head = 0;
#endif
return head; return head;
} }
...@@ -184,7 +190,13 @@ HOSTDEVICE int64_t indexer<0>(const Dim<0>& dim, int idx) { ...@@ -184,7 +190,13 @@ HOSTDEVICE int64_t indexer<0>(const Dim<0>& dim, int idx) {
#else #else
PADDLE_ASSERT(false); PADDLE_ASSERT(false);
#endif #endif
#if (defined __CUDA_ARCH__) && (CUDA_VERSION < 8000)
// On CUDA versions previous to 8.0, only __shared__ variables
// could be declared as static in the device code.
int64_t head = 0;
#else
static int64_t head = 0; static int64_t head = 0;
#endif
return head; return head;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册