MegEngine

HighLight

  • 此版本及后续版本将不再支持 cuda10.1。

Bugfix

第三方工具

  • 支持 atlas 输入 batch 为静态而输出 batch 为动态的情况;修复在多模型多线程的情况下 atlas 报 event 资源不够的问题。

通用组件

  • 优化 device 检查错误时的报错信息,使其更明确。
  • 修复 ConvLike 和 AxisAddRemove 的组合对 shape 错误处理导致报错的问题。

混合精度

  • 修复部分算子显示指定输入为 nchw format,例如 inp=tensor(a, format="nchw"),时出错的问题。

New Features

Python API

  • 增加对 cross 算子的支持。
  • 增加对高斯、泊松、拉普拉斯噪声算子(AdditiveGaussianNoise、AdditivePoissonNoise、AdditiveLaplaceNoise) 的支持。

通用组件

  • 添加 emboss, sharpen, linearcontrast 算子。

XLA

  • 添加 normalization 相关算子的 XLA lowering rule(目前支持batch norm, layer norm,group norm,instance norm)。
  • 添加 sort/argsort/topK 等排序相关算子的 XLA lowering rule。

Improvements

文档

  • 添加 复数、添加 DLPack 相关接口的使用说明。

MegEngine Lite

Bug Fixes

周边工具

  • 修复 MegEngine Lite 调用 get_io_tensor 时概率性获取到错误 device 类型的问题。

MegEngine

HighLight

  • This version and subsequent versions no longer support cuda10.1

Bugfix

Third party tools

  • Support the scenario where the input batch is static while the output batch is dynamic for Atlas, and fix the issue of insufficient event resources reported by Atlas in the case of multiple models and multiple threads.

Common components

  • Optimize error messages for device check errors.
  • Fix the combination of Convlike and Axisaddremove.

Mixed precision

  • Fixed an error in partial operator display when specifying input as nchw format, such as inp=tensor(a, format="nchw").

New Features

Python API

  • Add cross operator.
  • Support gaussian, laplace and poisson noise operator.

Common components

  • Add Operators include emboss,sharpen,linearcontrast.

XLA

  • Add XLA lowering rules for normalization ops.
  • Add xla lowering rules for sort/argsort/topK ops.

Improvements

Documentation

  • Add docstring for complex APIs and dlpack.

MegEngine Lite

Bug Fixes

Peripheral tools

  • Fix the issue in MegEngine Lite where there is a probability of obtaining an incorrect device type when calling get_io_tensor.

项目简介

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

🚀 Github 镜像仓库 🚀

源项目地址

https://github.com/MegEngine/MegEngine

发行版本 37

MegEngine v1.13.1

全部发行版

贡献者 39

全部贡献者

开发语言

  • C++ 79.8 %
  • Cuda 13.8 %
  • Python 4.9 %
  • C 0.9 %
  • CMake 0.5 %