首页
/ Tenstorrent Metal v0.59.0-rc15 版本技术解析

Tenstorrent Metal v0.59.0-rc15 版本技术解析

2025-07-10 19:16:05作者:农烁颖Land

Tenstorrent Metal 是一个面向高性能AI计算的硬件加速框架,专注于为深度学习工作负载提供高效的硬件支持。最新发布的v0.59.0-rc15版本带来了多项重要改进和功能增强,特别是在设备管理、性能优化和模型支持方面。

核心架构改进

本次版本在底层架构上进行了多项重要优化。首先是对设备初始化的重构,将固件构建和内存清除操作从设备初始化阶段移至MetalContext初始化阶段,这一改变显著提升了设备启动效率。同时改进了设备关闭流程,确保CloseDevice和CloseDevices使用相同的步骤序列,增强了系统稳定性。

在内存管理方面,移除了主机端缓冲区分配/释放的概念,转而采用更高效的transform接口来隐藏主机缓冲区操作细节。这一变化简化了API设计,同时提高了内存访问效率。全局循环缓冲区实现文件也被合并,减少了代码复杂度。

性能优化与硬件适配

针对Blackhole架构进行了多项适配性改进,包括修复了以太网微基准测试的挂起问题,调整了Argmax操作的每核心单元数以适应NOC宽度。特别值得注意的是增加了跟踪缓冲区大小,为性能分析和调试提供了更丰富的数据支持。

在路由优化方面,实现了对2D Fabric的动态路由支持,并优化了intermesh路由到下一个mesh的性能。新增了FabricContext组件,改进了设备初始化流程,支持在TG网关上启动fabric。

模型与算子增强

本次版本对多个深度学习模型提供了更好的支持。Mobilenetv2和Yolov8x等计算机视觉模型的演示程序得到了改进和修复。在自然语言处理方面,Llama模型的支持进一步增强,包括对Llama-3.1-8B-Instruct模型的精度调整。

算子层面新增了对uint16数据类型的bitwise或/异或操作支持,改进了Topk算子以支持子核心网格布局,并调整了Argmax算子基于NOC宽度的单元分配。卷积转置操作现在支持权重准备功能,扩展了模型构建的灵活性。

开发工具与测试改进

测试基础设施得到显著增强,新增了多种压力测试,包括多设备Eltwise和TM操作测试、连接开/关压力测试等。改进了noc状态检查脚本,为硬件调试提供了更强大的工具支持。

在CI/CD方面,优化了Docker代理使用,调整了测试目标设备选择策略,确保测试覆盖更全面。文档系统也进行了更新,改进了入门指南和API文档,帮助开发者更快上手。

这个版本标志着Tenstorrent Metal在稳定性、性能和功能完备性上的重要进步,为AI加速提供了更强大、更可靠的硬件支持平台。

登录后查看全文
热门项目推荐
相关项目推荐