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

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

2025-07-10 08:52:36作者:晏闻田Solitary

Tenstorrent TT-Metal 是一个专注于高性能计算和人工智能加速的开源项目,它提供了针对特定硬件架构优化的计算框架和工具链。本次发布的 v0.59.0-rc12 版本带来了多项重要改进和新功能,主要集中在性能优化、错误修复和新特性支持等方面。

核心架构改进

本次版本在底层架构方面进行了多项重要优化。首先是对设备初始化的重构,将固件构建和内存清理操作从设备初始化阶段移至MetalContext初始化阶段,这一改动显著提升了设备启动效率。同时,项目团队对设备池初始化移除了noexcept限定,使得错误处理更加灵活可靠。

在内存管理方面,版本引入了对ND分片(多维分片)的支持,为mesh设备和缓冲区提供了更灵活的内存分配策略。此外,项目团队还优化了主机端缓冲区操作,将其隐藏在transform接口之后,提升了代码的安全性和可维护性。

性能优化与功能增强

计算性能方面,本次版本对多个关键算子进行了优化。Topk算子现在支持子核心网格(sub_core_grid)并充分利用列中的可用核心,显著提升了处理能力。Argmax算子则根据NOC(片上网络)宽度调整了每核心处理单元数,实现了更好的资源利用率。

在张量操作方面,版本修复了Untilize操作在每核心输出通道大于256时的处理问题,并改进了分片行主序嵌入的支持。同时,项目团队还优化了批处理转置操作在tiled ttnn.concat中的使用,提升了拼接操作的效率。

新特性与模型支持

本次版本为多个流行模型提供了更好的支持。Mobilenetv2和Yolov9c模型被正式引入模型演示集,同时项目团队修复了Yolov8x演示中的问题。在生成模型方面,版本集成了VAE解码器到SDv1-4演示中,并更新了SDXL演示。

特别值得一提的是,项目团队为Llama-3.1-8B-Instruct模型添加了"performance"解码器精度覆盖,并优化了文本演示中的预取器性能模式。这些改进使得大模型推理更加高效稳定。

系统稳定性与测试增强

在系统稳定性方面,版本修复了Blackhole平台上的以太网微基准测试挂起问题,增加了跟踪缓冲区大小以捕获更多调试信息。项目团队还添加了多设备Eltwise和TM(张量操作)压力测试,以及连接打开/关闭压力测试,全面验证系统可靠性。

测试基础设施方面,版本引入了自动化GH工作流报告,优化了CI测试流程。特别针对Blackhole平台,项目团队调整了多个测试用例,确保在不同硬件配置下都能稳定运行。

开发者体验改进

对于开发者而言,本次版本带来了多项便利性改进。项目团队清理了多个头文件的包含关系,移除了未使用的文件,并重构了多个实现细节。在API方面,项目标准化了命名规范,将SLAVE改为更准确的SUBORDINATE表述。

文档方面也得到了加强,特别是对NOC API的测试套件文档进行了更新,使得开发者能更轻松地理解和使用这些底层接口。

总结

Tenstorrent TT-Metal v0.59.0-rc12版本在性能、稳定性和功能丰富度方面都取得了显著进步。从底层架构优化到上层模型支持,从核心计算能力提升到开发者体验改善,这个版本为高性能AI计算提供了更加坚实的基础。特别值得关注的是其对大模型推理的持续优化和对新兴硬件平台的适配,展现了项目团队对前沿技术趋势的敏锐把握。

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