Skip to content

v0.2.2 Release Notes

Compare
Choose a tag to compare
@MiaoYYu MiaoYYu released this 28 Feb 05:57
· 281 commits to main since this release
03056c2

Release Note

大家好!我们很高兴的宣布DIOPI v0.2.2版本正式发布!
DIOPI是一套设备无关接口体系,作为芯片于深度学习框架适配的“桥梁”。旨在训练框架和人工智能芯片之间定义一套计算契约,良好的函数抽象使得上(框架)下(芯片)两层在适配工程实施时能有效地解耦。基于这套契约,芯片开发者可以将下层芯片适配的工作复用到不同的训练框架(Pytorch、Parrots等)的适配中去,降低芯片+框架的适配成本,保障算子实现正确性。

版本号

v0.2.2

新增功能

  • 添加mmcv diopi测试;
  • 支持算子多级调用;
  • 支持3款芯片上的LLaMA 7B推理,2款芯片上的LLaMA 65B 推理;
  • 支持2款芯片上的InternLM 7B、65B推理和训练;

功能优化

  • diopi ci 生成数据的gen_data优化,不必每次运行ci都重新生成, (ci时间能缩短1半,磁盘占用能减少90%);
  • diopi test功能以及模块优化;

文档