cotengra icon indicating copy to clipboard operation
cotengra copied to clipboard

Best practice on running cotengra via the opt_einsum API

Open rht opened this issue 2 years ago • 1 comments

In https://github.com/dgasmith/opt_einsum/issues/217#issuecomment-1620525115, @jcmgray stated that running cotengra optimization via oe.contract_path(expression, *operands, optimize=opt) (where opt is a cotengra optimizer) is slower than doing it via quimb. I have to add more detail that the path finding part of the opt_einsum method alone is much slower than the entirety of the run via quimb. As such, the reasoning in that comment applies only to the contraction phase.

What is the recommended way to do the path finding via opt_einsum, that is performant? The main use case is that most circuits are written in Qiskit/Cirq, and with cuQuantum's CircuitToEinsum, it enables one to do contraction of any Qiskit/Cirq circuits.

rht avatar Jul 06 '23 10:07 rht

Hi @rht, to be clear cotengra and opt_einsum offer different optimizers that will take different amounts of time. But the same whether you call them via opt_einsum or quimb. However,

1 cotengra has more advanced optimizers, that can find better paths for large contractions. 2. For a given path, actually performing the contraction with cotengra can also be much faster as it uses batch matrix multiplication for hpyer indices, (but they will be similar for non-hyper contractions)

If you have the einsum equation and arrays, you can use contegra or quimb directly.

E.g.:

expr = ctg.contraction_expression(eq, *shapes, optimize=opt)
out = expr(*arrays)

jcmgray avatar Jul 10 '23 04:07 jcmgray