Skip to content

[WIP] feat(dcu): add BW1000 DCU backend support - #199

Open
chenyou74 wants to merge 4 commits into
InfiniTensor:masterfrom
chenyou74:feat/dcu-bw1000-master
Open

[WIP] feat(dcu): add BW1000 DCU backend support#199
chenyou74 wants to merge 4 commits into
InfiniTensor:masterfrom
chenyou74:feat/dcu-bw1000-master

Conversation

@chenyou74

Copy link
Copy Markdown

Summary

This PR adds Hygon BW1000/DCU backend support.
Main changes:

  • Add USE_DCU / USE_RCCL build options.
  • Add DCU runtime and device guard implementation.
  • Add RCCL communication backend.
  • Add DCU HIP kernels.
  • Enable GPT-2 and LLaMA3 examples to run with --device dcu.

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