diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/cntk/build.patch | 15 | ||||
| -rw-r--r-- | var/spack/repos/builtin/packages/cntk/package.py | 8 |
2 files changed, 6 insertions, 17 deletions
diff --git a/var/spack/repos/builtin/packages/cntk/build.patch b/var/spack/repos/builtin/packages/cntk/build.patch index 0b1e46e911..7ae6101db3 100644 --- a/var/spack/repos/builtin/packages/cntk/build.patch +++ b/var/spack/repos/builtin/packages/cntk/build.patch @@ -244,21 +244,6 @@ index 2c2e5cee7..c9c0072c9 100755 function find_gdk_include () { find_dir "$default_gdk_includes" "$gdk_include_check" -@@ -352,11 +382,14 @@ function show_help () - echo " --with-openblas[=directory] (experimental) $(show_default $(find_openblas))" - echo " --with-buildtype=(debug|release) $(show_default $default_buildtype)" - echo " --with-kaldi[=directory] $(show_default $(find_kaldi))" -+ echo " --with-openfst[=directory] $(show_default $(find_openfst))" - echo " --with-opencv[=directory] $(show_default $(find_opencv))" - echo " --with-libzip[=directory] $(show_default $(find_libzip))" - echo " --with-code-coverage[=(yes|no)] $(show_default ${default_use_code_coverage})" - echo " --with-boost[=directory] $(show_default $(find_boost))" - echo " --with-protobuf[=directory] $(show_default $(find_protobuf))" -+ echo " --with-multiverso[=directory] $(show_default $(find_multiverso))" -+ echo " --with-1bitsgd[=directory] $(show_default $(find_onebitsgd))" - echo " --with-py-versions=(space-separated list of 27, 34, 35)" - echo " --with-py27-path[=directory] $(show_default $(find_python 27))" - echo " --with-py34-path[=directory] $(show_default $(find_python 34))" @@ -751,6 +784,46 @@ do fi fi diff --git a/var/spack/repos/builtin/packages/cntk/package.py b/var/spack/repos/builtin/packages/cntk/package.py index 801c69dae4..025498fa8a 100644 --- a/var/spack/repos/builtin/packages/cntk/package.py +++ b/var/spack/repos/builtin/packages/cntk/package.py @@ -31,10 +31,10 @@ class Cntk(Package): via a directed graph.""" homepage = "https://www.microsoft.com/en-us/research/product/cognitive-toolkit" - url = "https://github.com/Microsoft/CNTK/archive/v2.0.beta15.0.tar.gz" + url = "https://github.com/Microsoft/CNTK/archive/v2.0.tar.gz" version('master', git='https://github.com/Microsoft/CNTK.git', branch='master') - version('2.0.rc1', url='https://github.com/Microsoft/CNTK/archive/v2.0.rc1.tar.gz') + version('2.0', '8038780f1169ceea578e5ef4d69e4c6f') variant('opencv', default=False, description="Enable OpenCV support.") variant('kaldi', default=False, description="Enable Kaldi support.") @@ -106,6 +106,10 @@ class Cntk(Package): args.append('--with-cudnn={0}' .format(spec['cudnn'].prefix)) args.append('--with-nccl={0}'.format(spec['nccl'].prefix)) + args.append('--with-gdk-include={0}' + .format(spec['cuda'].prefix.include)) + args.append('--with-gdk-nvml-lib={0}/stubs' + .format(spec['cuda'].prefix.lib64)) configure(*args) |