Skip to content

Commit 473eaa0

Browse files
authored
Merge pull request #3303 from yqshao/tensorflow-214_215-updates
update custom easyblock for Tensorflow for versions 2.14 + 2.15
2 parents 3d0a948 + 5467de5 commit 473eaa0

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
@@ -188,7 +188,7 @@ def is_version_ok(version_range):
188188
('gast', '2.0.0:'): 'gast_archive',
189189
('google.protobuf', '2.0.0:'): 'com_google_protobuf',
190190
('keras_applications', '2.0.0:2.2.0'): 'keras_applications_archive',
191-
('opt_einsum', '2.0.0:'): 'opt_einsum_archive',
191+
('opt_einsum', '2.0.0:2.15.0'): 'opt_einsum_archive',
192192
('pasta', '2.0.0:'): 'pasta',
193193
('six', '2.0.0:'): 'six_archive', # Part of Python EC
194194
('tblib', '2.4.0:'): 'tblib_archive',
@@ -608,6 +608,13 @@ def configure_step(self):
608608
# SYCL support removed in 2.4
609609
if LooseVersion(self.version) < LooseVersion('2.4'):
610610
config_env_vars['TF_NEED_OPENCL_SYCL'] = '0'
611+
# Clang toggle since 2.14.0
612+
if LooseVersion(self.version) > LooseVersion('2.13'):
613+
config_env_vars['TF_NEED_CLANG'] = '0'
614+
# Hermietic python version since 2.14.0
615+
if LooseVersion(self.version) > LooseVersion('2.13'):
616+
pyver = det_python_version(self.python_cmd)
617+
config_env_vars['TF_PYTHON_VERSION'] = '.'.join(pyver.split('.')[:2])
611618

612619
if self._with_cuda:
613620
cuda_version = get_software_version('CUDA')

0 commit comments

Comments
 (0)