Merge pull request #11175 from Flamefire/20200828161243_new_pr_Tensor…

…Flow220 patch out bug in collective primitive in TensorFlow 2.2.0
easybuilders · Sep 25, 2020 · 3b7914a · 3b7914a
2 parents 86ca97e + 584e28b
commit 3b7914a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
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
@@ -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
@@ -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',
         ],
     }),
 ]