f-dangel / backpack / 5383421886

98%

master: 98%

push

github

[REQ] Support PyTorch 2.x (#307)

* [REQ] Remove upper version restrictions for `torch` and `torchvision`

* [REQ] Bump python to 3.8+

* [REF] Replace `Tensor.symeig` with `torch.linalg.eigh`

* [CI] Replace `python3.7` with `python3.8`

* [REF] Try fixing syntax for `flake8` in `setup.cfg`

* [TEST] Skip double-backward of LSTM for PyTorch2.0.1

See https://github.com/pytorch/pytorch/issues/99413

* [FIX] flake8

* [TEST] Skip `jac_mat_prod` for LSTM in PyTorch2.0.1

double-backward not supported https://github.com/pytorch/pytorch/issues/99413

* [CI] Use python3.8 in RTD build

* [CI] Skip LSTM for PyTorch2.0.1 in DiagGGN tests

* [FIX] Imports

* [FIX] Turn off MKLDNN in RNN example

---------

Co-authored-by: Felix Dangel <felix.dangel@vectorinstitute.ai>

2 of 2 new or added lines in 1 file covered. (100.0%)

4415 of 4486 relevant lines covered (98.42%)

13.72 hits per line