3 ms·

I just compiled tensorflow on amd epyc and had no idea https://github.com/oneapi-src/oneDNN https://github.com/oneapi-src/oneDNN was actually MKL... now Im wond

by ramoz 6y ago

I just compiled tensorflow on amd epyc and had no idea https://github.com/oneapi-src/oneDNN https://github.com/oneapi-src/oneDNN was actually MKL... now Im wondering if I even am getting all that power

jcranmer 6y ago
The actual cpuid checking code is drilled from here: https://github.com/oneapi-src/oneDNN/blob/master/src/cpu/x64/cpu_isa_traits.hpp#L187 https://github.com/oneapi-src/oneDNN/blob/master/src/cpu/x64... to here: https://github.com/oneapi-src/oneDNN/blob/master/src/cpu/x64/xbyak/xbyak_util.h#L405 https://github.com/oneapi-src/oneDNN/blob/master/src/cpu/x64... It's using feature-flag checks, not family checks, so you shouldn't be affected if you're using oneDNN.
- ramoz 6y ago
  thank you!