Repository navigation
feat: 接入 Cambricon MLU 设备与 13 个 aten 算子 - #5
Conversation
- 新增 cambricon 设备 profile(vendor: cambricon,dispatch_key PrivateUse1) 与对应环境锁 configs/env/cambricon.lock.yaml - common 适配 MLU:device.py 顶部提前 import torch(避免 triton 命名空间重复注册)、 perf.py 补 torch.mlu.synchronize()(原只同步 CUDA)、perf_registry 登记 13 个 ops provider - 新增 perf/baselines/cambricon.json(v26.08.0 容器栈实测,31 用例; 环境升级后有意重采) - ops/ 新增 13 个算子:type_as / clone / contiguous / copy_ / dropout(Triton 级) + empty / empty_like / empty_strided / detach / detach_ / item / _local_scalar_dense / result_type(torch 级),三层全绿并在 ops/README.md 登记 环境: torch 2.12.1+cpu / torch_mlu 1.34.1 / triton 3.4.0+mlu2.1.3 / flag_gems 5.0.2
并行开发导致的登记类冲突,双方内容并集保留: - common/perf_registry.py:main 的 ops.sdpa / ops.embedding 与本分支 13 个 MLU 算子 provider 共 15 条并列注册 - ops/README.md:算子索引表保持连续(header + sdpa/embedding + 13 个 MLU 行),MLU 说明块与既有说明块均保留在表后 本地已复现 CI 全部步骤:链接/锚点、Python 语法、YAML、性能基线结构、 intake 契约 + pytest 20 passed
|
📦 后续维护提示(#16 审计整改关联,涉及你在 PR#5 引入的 MLU 算子族): ops/ 全库审计后,你当初引入的算子目录(clone/contiguous/copy_/detach/dropout/empty 系/item/result_type/type_as/_local_scalar_dense)已在 #16 统一补齐包化入口 后续这些算子做平台扩展时注意:①合并前照 templates/operator/README「交付自查清单」打勾;②多平台化后补 PLATFORM.md/MERGE.md(范式:ops/embedding 双份);③ |
收到,谢谢 🙏
|
环境: torch 2.12.1+cpu / torch_mlu 1.34.1 / triton 3.4.0+mlu2.1.3 / flag_gems 5.0.2