Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor: Use memory_op to set diag_const_nums #5246

Merged
merged 12 commits into from
Oct 18, 2024

Conversation

Critsium-xy
Copy link
Collaborator

Using memory_op to set diag_const_nums allows diag consts being directly allocated to some hardwares like dsp. Most hardwares don't need this because they will automatically copy the consts to their device, but some will need you to do this manually. This is a way to solve this issue. It is a foundation for transplanting abacus to dsp hardware( or some similar hardwares later ). This has not influence on devices like gpu and dcu.

@mohanchen mohanchen added the GPU & DCU & HPC GPU and DCU and HPC related any issues label Oct 17, 2024
Copy link
Collaborator

@haozhihan haozhihan left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@mohanchen mohanchen merged commit 545d2ec into deepmodeling:develop Oct 18, 2024
14 checks passed
@Critsium-xy Critsium-xy deleted the consts_refactor branch October 18, 2024 05:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
GPU & DCU & HPC GPU and DCU and HPC related any issues
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants