Skip to content

Commit c67df83

Browse files
author
C3SE "c3-builder" bot
committed
Update tensorflow block for changes in 2.14 and 2.15
1 parent 3e3854a commit c67df83

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

easybuild/easyblocks/t/tensorflow.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def is_version_ok(version_range):
187187
('gast', '2.0.0:'): 'gast_archive',
188188
('google.protobuf', '2.0.0:'): 'com_google_protobuf',
189189
('keras_applications', '2.0.0:2.2.0'): 'keras_applications_archive',
190-
('opt_einsum', '2.0.0:'): 'opt_einsum_archive',
190+
('opt_einsum', '2.0.0:2.15.0'): 'opt_einsum_archive',
191191
('pasta', '2.0.0:'): 'pasta',
192192
('six', '2.0.0:'): 'six_archive', # Part of Python EC
193193
('tblib', '2.4.0:'): 'tblib_archive',
@@ -595,6 +595,13 @@ def configure_step(self):
595595
# SYCL support removed in 2.4
596596
if LooseVersion(self.version) < LooseVersion('2.4'):
597597
config_env_vars['TF_NEED_OPENCL_SYCL'] = '0'
598+
# Clang toggle since 2.14.0
599+
if LooseVersion(self.version) > LooseVersion('2.13'):
600+
config_env_vars['TF_NEED_CLANG'] = '0'
601+
# Hermietic python version since 2.14.0
602+
if LooseVersion(self.version) > LooseVersion('2.13'):
603+
pyver = det_python_version(self.python_cmd)
604+
config_env_vars['TF_PYTHON_VERSION'] = '.'.join(pyver.split('.')[:2])
598605

599606
if self._with_cuda:
600607
cuda_version = get_software_version('CUDA')

0 commit comments

Comments
 (0)