Skip to content

Vectorise `test_gcnn_equivariance`

Vicentini Filippo requested to merge github/fork/attila-i-szabo/gcnn-test into master

Created by: attila-i-szabo

This PR merges the symmetry-related input spin configurations in test_gcnn_equivariance into a single array so they can be pulled through the GCNN in a single call, as suggested in #1171 (closed). On my laptop, this leads to a factor of ~3 speedup.

I don't think there's a similar quick fix for test_gcnn, the long runtimes there probably have to do with jitting everything you need for VMC.

Merge request reports