diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/mxnet/package.py | 1 | ||||
| -rw-r--r-- | var/spack/repos/builtin/packages/mxnet/parallell_shuffle.patch | 13 |
2 files changed, 14 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/mxnet/package.py b/var/spack/repos/builtin/packages/mxnet/package.py index 2ff65265a7..8ef5035c07 100644 --- a/var/spack/repos/builtin/packages/mxnet/package.py +++ b/var/spack/repos/builtin/packages/mxnet/package.py @@ -46,6 +46,7 @@ class Mxnet(MakefilePackage): patch('makefile.patch', when='@0.10:0.11') patch('makefile.opencv.patch', when='@1.6.0') + patch('parallell_shuffle.patch', when='@1.6.0') def build(self, spec, prefix): # copy template configuration file diff --git a/var/spack/repos/builtin/packages/mxnet/parallell_shuffle.patch b/var/spack/repos/builtin/packages/mxnet/parallell_shuffle.patch new file mode 100644 index 0000000000..eb286cde57 --- /dev/null +++ b/var/spack/repos/builtin/packages/mxnet/parallell_shuffle.patch @@ -0,0 +1,13 @@ +--- a/src/operator/random/shuffle_op.cc 2020-01-31 08:45:22.000000000 +0900 ++++ b/src/operator/random/shuffle_op.cc 2020-06-15 11:34:32.000000000 +0900 +@@ -22,8 +22,8 @@ + * \file shuffle_op.cc + * \brief Operator to shuffle elements of an NDArray + */ +-#if !defined (__ANDROID__) && ((__GNUC__ > 4 &&\ +- !defined(__clang__major__)) || (__clang_major__ > 4 && __linux__)) ++#if ((__GNUC__ > 4 && !defined(__clang__major__)) || (__clang_major__ > 4 && __linux__)) && \ ++ defined(_OPENMP) && !defined(__ANDROID__) + #define USE_GNU_PARALLEL_SHUFFLE + #endif + |