f-dangel / curvlinops / 8364730614

90%

push

github

Allow for matrix-matrix and matrix-vector products with `KFACLinearOperator` and `KFACInverseLinearOperator` without converting to numpy (#91)

* Add tests for torch_matmat and torch_matvec

* Add torch_matmat and torch_matvec methods to KFACLinearOperator

* Nicer if-statement

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>

* Fix doc string

* Add tests for torch_matmat and torch_matvec for KFACInverseLinearOperator

* Add torch_matmat and torch_matvec to KFACInverseLinearOperator

* Increase test tolerance

* Deduce torch_matmat return type from input type

* Remove device check

* Make constant explicit

* Remove leftover import

* Add support for list inputs to torch_matvec

* Add tests for list inputs to torch_matmat and torch_matvec

* Address review feedback

* Switch to case for kfac tests and minor fixes

* Fix flake8

---------

Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>

52 of 58 new or added lines in 2 files covered. (89.66%)

1017 of 1138 relevant lines covered (89.37%)

0.89 hits per line