Skip to content

Conversation

@Qianruipku
Copy link
Collaborator

@Qianruipku Qianruipku commented Oct 29, 2024

Now init stochastic wf in CPU and then sync to GPU
Add template for sdft to support different devices.

Linked Issue

Part work of #5296

@Qianruipku Qianruipku changed the title Feature: make sdft support GPU Feature: make init stochastic WF support GPU Oct 30, 2024
@Qianruipku Qianruipku marked this pull request as ready for review October 30, 2024 13:33
@mohanchen mohanchen merged commit 4294ee7 into deepmodeling:develop Oct 31, 2024
14 checks passed
@mohanchen mohanchen added the Refactor Refactor ABACUS codes label Oct 31, 2024
Fisherd99 pushed a commit to Fisherd99/abacus-BSE that referenced this pull request Mar 31, 2025
* add bcast for energy

* add template <Device> for sdft

* fix compile

* fix compile without mpi

* add template for precision

* Make init stochastic WF support GPU

* fix compile without mpi

* fix compile after merge

* fix bug of init_com_orbitals
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Refactor Refactor ABACUS codes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants