Re:XLA: linear algebra library for TensorFlow
by acmj on Wednesday March 08, 2017 @09:22AM
(#13820382)
I later realized he is the developer, but this does not change this discussion. Here is a micro benchmark, computing softmax 1 million times over a random vector of size 1000. On an old Linux server, calling the libm expf once takes 11.76 CPU seconds; calling it twice takes 25.15s. The implementation for calling expf once:
This micro benchmark proves my point: using expf from libm, the reference implementation in nnpack is suboptimal. It is possible that a vectorized expf may change the fact, but the developer needs to prove it with numbers.
[ Parent
| on HN ]
There are no comments on this comment yet.