TensorFlow-1.0.0-CrayGNU-2016.11-Python-3.5.2.eb

# @author: marcelsc, gppezzi, victorusu

easyblock = "CmdCp"

name = 'TensorFlow'
version = '1.0.0'
pyver = '3.5.2'
versionsuffix = '-Python-%s' % pyver

homepage = 'https://www.tensorflow.org/'
description = """An open-source software library for Machine Intelligence."""

toolchain = {'name': 'CrayGNU', 'version': '2016.11'}

builddependencies = [
    ('wheel', '0.30.0a0', '-Python-%(pyver)s'),
    ('Bazel', '0.4.5', '-Python-%(pyver)s'),
]

dependencies = [
    ('Python', pyver),
    ('SWIG', '3.0.10', '-Python-%(pyver)s'),
    ('protobuf', '3.1.0', '-Python-%(pyver)s'),
]

sources = ['v%(version)s.tar.gz']
source_urls = ['https://github.com/tensorflow/tensorflow/archive/']

patches = [('tensorflow-v%(version)s-Python-%(pyver)s-cscs.patch')]

with_configure = False

build_str  = "export TEST_TMPDIR=/dev/shm/$USER/bazelout/ && "
# Exporting CC breaks this build on Cray
build_str += "unset CC && "
build_str += "chmod +x configure-cscs.sh && "
build_str += "./configure-cscs.sh && "
build_str += "bazel build --verbose_failures --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-msse4.2 --copt=-msse4.1 "
build_str += "-c opt //tensorflow/tools/pip_package:build_pip_package &&"
build_str += "bazel-bin/tensorflow/tools/pip_package/build_pip_package %(builddir)s &&"
build_str += "pip install --no-deps --prefix  %(builddir)s %(builddir)s/tensorflow-%(version)s-cp35-cp35m-linux_x86_64.whl"

cmds_map = [
    ('v%(version)s.tar.gz', build_str),
]

files_to_copy = [
	      (['%(builddir)s/tensorflow-%(version)s-cp35-cp35m-linux_x86_64.whl'], '%(installdir)s'),
	      (['%(builddir)s/bin'], '%(installdir)s'),
	      (['%(builddir)s/lib'], '%(installdir)s'),
]

sanity_check_paths={
    'files': ['%(builddir)s/tensorflow-%(version)s-cp35-cp35m-linux_x86_64.whl'],
    'dirs': ['%(installdir)s/lib/python%(pyshortver)s/site-packages'],
}

postinstallcmds = [ 'rm -rf /dev/shm/$USER/bazelout/' ]

modextrapaths = {                                                                  
    'PYTHONPATH' : 'lib/python%(pyshortver)s/site-packages',
}

moduleclass = 'devel'