We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0ad9b6c commit e020a96Copy full SHA for e020a96
1 file changed
eb_hooks.py
@@ -671,6 +671,12 @@ def parse_hook_tensorflow_h5py_glibc(ec, eprefix):
671
'EOF\n'
672
'python3 /tmp/fix_h5py.py && '
673
)
674
+ if get_eessi_envvar('EESSI_CPU_FAMILY') == 'aarch64':
675
+ ec['prebuildopts'] = ec.get('prebuildopts', '') + (
676
+ # KleidiAI in TF 2.18 has similar -march/-mcpu conflict as XNNPACK
677
+ # The easyblock already excludes XNNPACK from -mcpu=native, extend the same exclusion to KleidiAI
678
+ 'sed -i \'s|--per_file_copt=-.*XNNPACK/.*@-mcpu=native|--per_file_copt=-.*XNNPACK/.*,-.*KleidiAI/.*@-mcpu=native|g\' .tf_configure.bazelrc && '
679
+ )
680
current_opts = ec.get('buildopts', [])
681
if isinstance(current_opts, str):
682
current_opts = current_opts.split()
0 commit comments