easybuild-easyblocks
easybuild-easyblocks copied to clipboard
Add support for MKL/oneDNN in TensorFlow on Aarch64
TF 2.4 (not sure about exact version where this was introduced) has an option --config=mkl_aarch64.
Currently we don't use with_mkl_dnn EC param on non-x86 so this could be extended to aarch64.
We also add --config=mkl when mkl-dnn is a dependency. This may or may not work for aarch64 and should be checked.