search

mind / wheels

Performance-optimized wheels for TensorFlow (SSE, AVX, FMA, XLA, MPI)
886 stars 108 forks source link

AVX512F instruction support #8

Open thupalo opened 6 years ago

thupalo commented 6 years ago

Thank you very much for great wheels. Recently I upgraded TF 1.3->1.4 GPU on i9 7900x CPU, with no major issues. (Intel Math Kernel Library had to be installed and numpy upgrade - but OK)

May I ask for some new addition regarding the info message I had "Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX512F" any plans for this? Thank you once again and best regards / thupalo

danqing commented 6 years ago

Congrats for getting a great CPU!

This will definitely be supported sometime in the future. We may be able to provide CPU wheels soon as I suspect AWS C5 instances have AVX512, but GPU wheels may not happen for a while. This new instruction set is only supported by high end Skylake CPUs, which we don't physically have ourselves and are not available in any cloud offerings (that also provide GPU).

ghost commented 6 years ago

When can this be supported? AWS new T3 instances all come with Intel Scalable Xeon processors, thus AVX512.