Skip to content

ditorch v0.1.0 releasenote

Choose a tag to compare

@MiaoYYu MiaoYYu released this 02 Sep 07:45
· 50 commits to main since this release
096eecb

大家好!我们很高兴的宣布ditorch仓库正式开源,并发布v0.1.0版本。

ditorch 是设备无关 torch, 旨在屏蔽各硬件厂商 torch 差异,为用户提供一致使用体验。通过 ditorch,开发者可以适配多个硬件算子库;此外,ditorch 提供训练过程中需要的基础工具,解决模型训练过程中出现的痛点问题。

版本

v0.1.0

核心功能

1. 可无感切换 pytorch 至国产芯片

2. 提供多个基础工具,解决训练过程的问题

序号 工具 简介
1 算子参数抓取工具 抓取模型真实训练过程中真实的输入输出
2 精度分析工具 进行离线和实时的精度分析
3 速度分析工具 可进行离线和实时的耗时分析,协助性能优化
4 算子 Fallback 可将指定、全部算子在设备上运行的操作 fallback 到 CPU 计算