[Numpy-discussion] Tensor Contraction (HPTT) and Tensor Transposition (TCL)

Stephan Hoyer shoyer at gmail.com
Wed Aug 16 11:38:24 EDT 2017


On Wed, Aug 16, 2017 at 2:39 AM, Paul Springer <pavdev at gmx.de> wrote:

>
> What version of Numpy are you comparing to? Note that in 1.13 you can
> enable some optimization in einsum, and the coming 1.14 makes that the
> default and uses CBLAS when possible.
>
> I was using 1.10.4; however, I am currently running the benchmark with
> 1.13.1 and 'optimize=True'; this, however, seems to yield even worse
> performance (see attached).
> If you are interested, you can check the performance difference yourself
> via: ./benchmark/python/bechmark.sh
>

This sounds like you may be using relatively small matrices, where the
overhead of calculating the optimal strategy dominates. Can you try with a
few bigger test cases?
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20170816/687292f6/attachment.html>


More information about the NumPy-Discussion mailing list