Commit 7f6f35e
committed
feat(cambricon): enable MLU590 (device map + graph class)
Add the two touchpoints needed to bring up vllm-plugin-FL on Cambricon
MLU (torch_mlu), verified E2E with Qwen3-8B on MLU590:
- VENDOR_DEVICE_MAP: register cambricon -> mlu, so the vendor-device
lookup no longer raises ValueError for cambricon.
- Graph: select torch.mlu.MLUGraph for device_type == "mlu", so graph
capture no longer raises NotImplementedError("not support graph").
No dedicated cambricon dispatch backend is required; ops route through
flag_gems (Triton) generically. This is minimal enablement only.
This PR was written in part with the assistance of generative AI.1 parent f4ebc25 commit 7f6f35e
2 files changed
Lines changed: 4 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
51 | 51 | | |
52 | 52 | | |
53 | 53 | | |
| 54 | + | |
| 55 | + | |
54 | 56 | | |
55 | 57 | | |
56 | 58 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
51 | 51 | | |
52 | 52 | | |
53 | 53 | | |
| 54 | + | |
| 55 | + | |
54 | 56 | | |
55 | 57 | | |
56 | 58 | | |
| |||
0 commit comments