Closed hatemhelal closed 1 year ago
Check out this pull request on
See visual diffs & provide feedback on Jupyter Notebooks.
Powered by ReviewNB
I didn't spot this before but it looks like from the test timing report copied in the PR summary that the nuclear attraction integrals get slower with the switch to the series implementation of gammanu
I didn't spot this before but it looks like from the test timing report copied in the PR summary that the nuclear attraction integrals get slower with the switch to the series implementation of
gammanu
Interesting, and what happened to water[False]?
Still net goodness for CPU-based CI, and we know IPU will require some additional work.
what happened to water[False]?
I think this is caching / jit helping since we effectively compute the same matrix elements twice. For example, selecting just that test with -k test_water_eri
we see:
================================================== slowest 8 durations ==================================================
6.41s call test/test_experimental.py::test_water_eri[True]
1.94s call test/test_experimental.py::test_water_eri[False]
and -k test_water_eri[False]
runs in around 7 sec since it doesn't benefit from the True
case running ahead of it.
This PR adds another implementation
gammanu
to make use the series expansion of the lower incomplete gamma function. This is derived in the notebook included in this PR.Closes #116 which motivated this journey into obscure numerical methods.
Using the command
pytest --durations=8
will report the 8 slowest testpoints:CPU tests with
gammanu = gammanu_gamma
CPU tests with
gammanu = gammanu_series