Skip to content

Commit 77651f1

Browse files
authored
Merge pull request #1034 from trz42/2023.06-a64fx-2023a-eb482-apps-tf
{2023.06}[2023a,a64fx] add TensorFlow 2.13
2 parents 810ae1f + 0630587 commit 77651f1

2 files changed

Lines changed: 14 additions & 8 deletions

File tree

easystacks/software.eessi.io/2023.06/a64fx/eessi-2023.06-eb-4.9.4-2023a.yml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,13 +45,13 @@ easyconfigs:
4545
- nsync-1.26.0-GCCcore-12.3.0.eb
4646
- RE2-2023-08-01-GCCcore-12.3.0.eb
4747
- protobuf-python-4.24.0-GCCcore-12.3.0.eb
48-
## originally built with EB 4.8.2; PR 19268 included since EB 4.9.0
49-
## - TensorFlow-2.13.0-foss-2023a.eb:
50-
## # patch setup.py for grpcio extension in TensorFlow 2.13.0 easyconfigs to take into account alternate sysroot;
51-
## # see https://github.com/easybuilders/easybuild-easyconfigs/pull/19268
52-
## options:
53-
## from-pr: 19268
54-
# - TensorFlow-2.13.0-foss-2023a.eb
48+
# originally built with EB 4.8.2; PR 19268 included since EB 4.9.0
49+
# - TensorFlow-2.13.0-foss-2023a.eb:
50+
# # patch setup.py for grpcio extension in TensorFlow 2.13.0 easyconfigs to take into account alternate sysroot;
51+
# # see https://github.com/easybuilders/easybuild-easyconfigs/pull/19268
52+
# options:
53+
# from-pr: 19268
54+
- TensorFlow-2.13.0-foss-2023a.eb
5555
- X11-20230603-GCCcore-12.3.0.eb
5656
# originally built with EB 4.8.2; PR 19339 included since EB 4.9.0
5757
# - HarfBuzz-5.3.1-GCCcore-12.3.0.eb:

eb_hooks.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,13 @@ def post_ready_hook(self, *args, **kwargs):
132132
memory_hungry_build_a64fx = cpu_target == CPU_TARGET_A64FX and self.name in ['Qt5', 'ROOT']
133133
if memory_hungry_build or memory_hungry_build_a64fx:
134134
parallel = self.cfg['parallel']
135-
if parallel > 1:
135+
if cpu_target == CPU_TARGET_A64FX and self.name in ['TensorFlow']:
136+
# limit parallelism to 8, builds with 12 and 16 failed on Deucalion
137+
if parallel > 8:
138+
self.cfg['parallel'] = 8
139+
msg = "limiting parallelism to %s (was %s) for %s on %s to avoid out-of-memory failures during building/testing"
140+
print_msg(msg % (self.cfg['parallel'], parallel, self.name, cpu_target), log=self.log)
141+
elif parallel > 1:
136142
self.cfg['parallel'] = parallel // 2
137143
msg = "limiting parallelism to %s (was %s) for %s to avoid out-of-memory failures during building/testing"
138144
print_msg(msg % (self.cfg['parallel'], parallel, self.name), log=self.log)

0 commit comments

Comments
 (0)