Skip to content

Optimize density evaluation in PBC multigrid v3. - #891

Open
sunqm wants to merge 14 commits into
pyscf:masterfrom
bytedance:optimize-eval_density
Open

Optimize density evaluation in PBC multigrid v3.#891
sunqm wants to merge 14 commits into
pyscf:masterfrom
bytedance:optimize-eval_density

Conversation

@sunqm

@sunqm sunqm commented Sep 2, 2026

Copy link
Copy Markdown
Collaborator
  • evaluate_density and evaluate_tau CUDA kernels are optimized.
  • Update FFT bucket construction and mesh assignment to improve numerical consistency during PES scans.
  • The previous density implementation based on multigrid_v2 is used as a fallback implementation.
  • multigrid-v3 is the default in pip wheel. multigrid-v2 kernels in prebuilt wheels is disabled.
  • Other parameter adjustments for numerical stability and performance improvements

@sunqm
sunqm marked this pull request as ready for review September 5, 2026 01:43
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