寒武纪：咱们是否能支持寒武纪？

[y149604146]

Reminder

• I have read the README and searched the existing issues.

System Info

寒武纪驱动信息：
Wed Dec 25 00:03:02 2024
+------------------------------------------------------------------------------+
| CNMON v5.10.10 Driver v5.10.10 |
+-------------------------------+----------------------+-----------------------+
| Card VF Name Firmware | Bus-Id | Util Ecc-Error |
| Fan Temp Pwr:Usage/Cap | Memory-Usage | SR-IOV Compute-Mode |
|===============================+======================+=======================|
| 0 / MLU370-X4 v1.1.6 | 0000:34:00.0 | 0% N/A |
| 0% 30C 27 W/ 150 W | 0 MiB/ 23308 MiB | N/A Default |
+-------------------------------+----------------------+-----------------------+

+------------------------------------------------------------------------------+
| Processes: |
| Card MI PID Command Line MLU Memory Usage |
|==============================================================================|
| No running processes found |
+------------------------------------------------------------------------------+

寒武纪迁移文档：https://icnv0j6dma9x.feishu.cn/wiki/M0I5wUbYRiHsKqkfMUncEsZ9npf
我理解在系统开始的时候判别一下系统版本，说为寒武纪，则用所有import pytorch的地方加上两行代码。
import torch_mlu
from torch_mlu.utils.model_transfer import transfer

Reproduction

咱们有计划supoort 寒武纪吗？多谢

Expected behavior

支持寒武纪的MLU

Others

期待你的回复，没有plan也希望有个回复，如果有好的建议就更好了，比如如果我个人改代码，改具体哪几个地方？（除了import torch的地方加上两行“ import torch_mlu
from torch_mlu.utils.model_transfer import transfer”）