You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by lx...@apache.org on 2017/07/07 15:58:22 UTC
[09/50] [abbrv] incubator-mxnet-test git commit: Don't use
__CUDACC_VER__ (#6844)
Don't use __CUDACC_VER__ (#6844)
Project: http://git-wip-us.apache.org/repos/asf/incubator-mxnet-test/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-mxnet-test/commit/49cd2e66
Tree: http://git-wip-us.apache.org/repos/asf/incubator-mxnet-test/tree/49cd2e66
Diff: http://git-wip-us.apache.org/repos/asf/incubator-mxnet-test/diff/49cd2e66
Branch: refs/heads/master
Commit: 49cd2e66941937791940efd50ec1e00db09b035e
Parents: 17620fe
Author: Przemyslaw Tredak <pt...@gmail.com>
Authored: Wed Jun 28 09:32:44 2017 -0700
Committer: Eric Junyuan Xie <pi...@users.noreply.github.com>
Committed: Wed Jun 28 09:32:44 2017 -0700
----------------------------------------------------------------------
src/operator/tensor/sort_op-inl.cuh | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/incubator-mxnet-test/blob/49cd2e66/src/operator/tensor/sort_op-inl.cuh
----------------------------------------------------------------------
diff --git a/src/operator/tensor/sort_op-inl.cuh b/src/operator/tensor/sort_op-inl.cuh
index 10ba61f..725f9b8 100644
--- a/src/operator/tensor/sort_op-inl.cuh
+++ b/src/operator/tensor/sort_op-inl.cuh
@@ -7,8 +7,8 @@
#define MXNET_OPERATOR_TENSOR_SORT_OP_INL_CUH_
#include <thrust/device_ptr.h>
#include <thrust/sort.h>
-#if defined(_MSC_VER) && __CUDACC_VER__ != 80044
-// Many CUDA compilers other than V8.0.44 crash on Windows
+#if defined(_MSC_VER) && __CUDACC_VER_MAJOR__ == 8 && __CUDACC_VER_BUILD__ != 44
+// Many CUDA 8 compilers other than V8.0.44 crash on Windows
#pragma warning("Potential crash on CUDA compiler detected. Switching sorting from CUB to Thrust")
#define SORT_WITH_THRUST
#else