li012589 / TRGbenchmark

Using simple TRG code to benchmark different tensor implements.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Benchmark of Different Tenosr Implements

Currently, this test focus on three different implements: using pytorch; using julia and using itensor.

The result is

Implement Time(average over 10 runs)
pytorch 1.73s
julia 9.31s
itensor 9.75s

The scaling curve with iterations is

scaling curve

The scaling curve with maximum cuts is

cut

Details

This test is done on a mid 2014 MacBook Pro with a 2.2GHz Intel i7 processor and 16G of memory. And the pytorch test doesn't utilize a GPU.

Library versions:

name version
pytorch 1.0.0.dev20180921(nightly)
julia 1.01
itensor 2.1.1

To perform this test on your machine, run

python ./test.py
python ./testscaling.py

About TRG Algorithm

A beautiful tutorial of TRG by itensor

About

Using simple TRG code to benchmark different tensor implements.


Languages

Language:Python 50.8%Language:C++ 24.1%Language:Julia 15.8%Language:Makefile 9.3%