diff --git a/easybuild/easyconfigs/t/TensorFlow/TensorFlow-2.2.0-foss-2019b-Python-3.7.4.eb b/easybuild/easyconfigs/t/TensorFlow/TensorFlow-2.2.0-foss-2019b-Python-3.7.4.eb index 7c4e03c0c7d..d40a0ab6902 100644 --- a/easybuild/easyconfigs/t/TensorFlow/TensorFlow-2.2.0-foss-2019b-Python-3.7.4.eb +++ b/easybuild/easyconfigs/t/TensorFlow/TensorFlow-2.2.0-foss-2019b-Python-3.7.4.eb @@ -129,6 +129,7 @@ exts_list = [ 'patches': [ 'TensorFlow-1.14.0_swig-env.patch', 'TensorFlow-2.1.0_fix-cuda-build.patch', + 'TensorFlow-2.1.0_fix-collective-all-reduce-strategy.patch', ], 'source_tmpl': 'v%(version)s.tar.gz', 'source_urls': ['https://github.com/tensorflow/tensorflow/archive/'], @@ -137,6 +138,8 @@ exts_list = [ '69cd836f87b8c53506c4f706f655d423270f5a563b76dc1cfa60fbc3184185a3', # v2.2.0.tar.gz 'b83cce6b91c7d19b8b320158ffc50fb4b2de454f5ac191c58d704234a1bf9005', # TensorFlow-1.14.0_swig-env.patch '78c20aeaa7784b8ceb46238a81e8c2461137d28e0b576deeba8357d23fbe1f5a', # TensorFlow-2.1.0_fix-cuda-build.patch + # TensorFlow-2.1.0_fix-collective-all-reduce-strategy.patch + '093f4dd3ec372a82d50dffe32eea6821025cd1c406911a746c4367a40bc38486', ], }), ] diff --git a/easybuild/easyconfigs/t/TensorFlow/TensorFlow-2.2.0-fosscuda-2019b-Python-3.7.4.eb b/easybuild/easyconfigs/t/TensorFlow/TensorFlow-2.2.0-fosscuda-2019b-Python-3.7.4.eb index fe727062968..a84d2cebb1f 100644 --- a/easybuild/easyconfigs/t/TensorFlow/TensorFlow-2.2.0-fosscuda-2019b-Python-3.7.4.eb +++ b/easybuild/easyconfigs/t/TensorFlow/TensorFlow-2.2.0-fosscuda-2019b-Python-3.7.4.eb @@ -131,6 +131,7 @@ exts_list = [ 'patches': [ 'TensorFlow-1.14.0_swig-env.patch', 'TensorFlow-2.1.0_fix-cuda-build.patch', + 'TensorFlow-2.1.0_fix-collective-all-reduce-strategy.patch', ], 'source_tmpl': 'v%(version)s.tar.gz', 'source_urls': ['https://github.com/tensorflow/tensorflow/archive/'], @@ -139,6 +140,8 @@ exts_list = [ '69cd836f87b8c53506c4f706f655d423270f5a563b76dc1cfa60fbc3184185a3', # v2.2.0.tar.gz 'b83cce6b91c7d19b8b320158ffc50fb4b2de454f5ac191c58d704234a1bf9005', # TensorFlow-1.14.0_swig-env.patch '78c20aeaa7784b8ceb46238a81e8c2461137d28e0b576deeba8357d23fbe1f5a', # TensorFlow-2.1.0_fix-cuda-build.patch + # TensorFlow-2.1.0_fix-collective-all-reduce-strategy.patch + '093f4dd3ec372a82d50dffe32eea6821025cd1c406911a746c4367a40bc38486', ], }), ]