Skip to content

Further screening in multigrid v2#697

Open
Walter-Feng wants to merge 18 commits intopyscf:masterfrom
Walter-Feng:multi-grid/extreme_screening
Open

Further screening in multigrid v2#697
Walter-Feng wants to merge 18 commits intopyscf:masterfrom
Walter-Feng:multi-grid/extreme_screening

Conversation

@Walter-Feng
Copy link
Contributor

The major changes are as follows:

a. Have a more fine-tuned screening of the gaussian pairs over the grid points;
b. Allow basis transformation between contracted and primitive in a block-sparse fashion;
c. A raw kernel in cupy for contraction of Gv with density / potential in Fourier space;
d. Storage of coulG to avoid redundant calculation per Fock build.

These factors contribute to the fast timing of Fock build, with around 2 ~ 3 times of speedup for water clusters.

Some notes:
a. I store the intermediate of local part of pseudopotential in object creation, but should be stored when get_pp is executed which already generates vpplocG. This part is not negligible in gradient evaluation.
b. This branch by itself does not allow full calculation of 512 water molecules for 80 GB of RAM. This is due to a change in overlap integral in master branch, which stores (ncell, ncart, ncart) as an intermediate. Branch multi-grid/publish in my fork does not contain such change and is able to run 512 water molecules on A100s.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant