Skip to content
Unverified Commit 0f70de03 authored by Andreas Klöckner's avatar Andreas Klöckner Committed by GitHub
Browse files

Add make_einsum (#328)



* Add make_einsum

* einsum: allow customizing arg names, specify name

* Fix double use of variable i in make_einsum

Co-authored-by: default avatarKaushik Kulkarni <15399010+kaushikcfd@users.noreply.github.com>

* Use dictionary unpacking for knl_creation_kwargs

* Add some einsum unit tests

* Fix up einsum tests

* Update comment: 'diagonalization' -> 'extract diagonal'

* Add documentation for "make_einsum"

* Fix doc build errors

* Check that the number of arg names matches the number of arg specs

* Add error handling test

* Update make_einsum docstring

* Compactify kernel creation

* Ensure output subscripts contain unique indices

* Add einsum sanity check in unit tests

* Use set comprehension in make_einsum

* Add note about lack of complexity reduction in make_einsum

* Remove whitespace in docstring (flake8)

* Fix doc syntax in make_einsum

Co-authored-by: default avatarKaushik Kulkarni <15399010+kaushikcfd@users.noreply.github.com>
Co-authored-by: default avatarThomas Gibson <gibsonthomas1120@hotmail.com>
parent dddf4e4c
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment