领域加速库

ATB加速库

• MLA PO算子泛化功能增强。补充滑动窗口、量化叠加Lse输出，headnum[16~128]16倍数功能支持。
• ATB支持单算子编译调试，支持单算子kernel printf和dump tensor。
• ATB支持通过增加IF NODE走入不同的构图，提升易用性。
• ATB Python API功能增强，支持设置显存空间大小。