rust icon indicating copy to clipboard operation
rust copied to clipboard

CPU version should be default compile with MKL

Open usamec opened this issue 6 years ago • 2 comments

Tensorflow compiled with MKL is usually much faster. We should have option for this. https://www.tensorflow.org/guide/performance/overview

usamec avatar Feb 06 '19 22:02 usamec

We can't compile with MKL by default, because that's processor-specific. You can compile with MKL by either setting the TF_RUST_BAZEL_OPTS environment variable to --config=mkl before building, or by compiling TensorFlow manually and installing it where pkg-config can find it.

adamcrume avatar Feb 08 '19 03:02 adamcrume

Great, can we have that env var setting in the documentation?

Although package feature flag would be even better, env var is enough. Thanks.

usamec avatar Feb 14 '19 10:02 usamec