CHANGELOG

CATLASS 2.X

CATLASS 2.0.0

CATLASS DSL

  • 正式上线CATLASS DSL支持Pythonic Kernel,原生支持Tile编程。在 AscendNPU-IR 的基础上构建 TLA Dialect,将 Python 中描述的 TLA 操作降级为 MLIR,并由 AscendNPU-IR 工具链编译为可执行的核函数产物。
    • 覆盖矩阵乘、向量计算、动态控制流等完整算子表达能力,tla.print 支持GM/UB上的格式化标量输出及动态控制流内 tensor/scalar 打印
    • Host 侧提供统一的 tensor 构造入口 from_dlpack / make_fake_tensor,支持 @dataclass 结构体作为 kernel 入参
  • 样例
    • Basic Mmad:基础矩阵乘示例,basic_matmul、basic_matmul_atomic_add、basic_matmul_mutex、basic_matmul_mutex_with、basic_mmad_ptr(mutex / atomic_add / ptr 多路径)
    • Basic Mixed:基础CV算子示例,basic_mixed、basic_mixed_mutex、basic_mixed_fixpipe_nz2dn、basic_mixed_store_zN、basic_mixed_store_zNUnAlign、basic_mixed_ub2l1(混合 copy+mmad 场景)
    • Matmul Epilogue:后处理类CV融合样例,matmul_add、matmul_add_ub、matmul_bias、matmul_leaky_relu、matmul_sigmoid、matmul_silu、matmul_tanh
    • StreamK Matmul:StreamK 调度矩阵乘,normal tile 全 K 计算 + 尾轮 tile 多核切 K 归约
    • Multi Core SplitK Matmul:多核切 K 矩阵乘,multi_core_splitk_matmul、tail_multi_core_splitk_matmul
    • Batched Matmul:批量矩阵乘
    • Grouped Matmul Slice M:按 M 轴切分的分组矩阵乘
    • Flash Attention Infer:推理Flash Attention样例demo,支持非Mask非Paged场景,支持对齐/非对齐输入
  • 工具支持
    • 构建体系:基于 AscendNPU-IR 子模块构建,提供 Docker 开发镜像,支持 CANN >= 9.1.0
  • 文档资料

CATLASS C++

CATLASS 1.X

CATLASS 1.6.0

CATLASS 1.5.0

CATLASS 1.4.0

CATLASS 1.3.0

CATLASS 1.2.0

CATLASS 1.1.0

CATLASS 1.0.0

  • CATLASS模板库正式开源发布

  • 关键特性

    • 提供Kernel、Block、Tile、Basic分层算子开发能力
  • 样例参考

    • 提供包括基础Matmul及各种不同的优化策略在内的算子样例