easybuild-easyblocks icon indicating copy to clipboard operation
easybuild-easyblocks copied to clipboard

Add support for MKL/oneDNN in TensorFlow on Aarch64

Open Flamefire opened this issue 4 years ago • 0 comments

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.

Flamefire avatar Jan 25 '21 16:01 Flamefire