pypto.frontend.function#

产品支持情况#

• Ascend 950PR：支持

• Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持

• Atlas A2 训练系列产品/Atlas A2 推理系列产品：支持

功能说明#

pypto.frontend.function用于定义可复用的计算子图或函数模块，允许在多个内核函数中复用相同的计算逻辑。该功能旨在提高代码的模块化和可维护性。

预期特性：

• 代码复用: 将常用的计算模式封装为函数

• 模块化设计: 构建复杂算子时可以组合多个小函数

• 类型安全: 函数签名中明确输入输出类型

• 自动优化: 编译器可以内联或优化函数调用

函数原型#

@pypto.frontend.function
def reusable_function(
    arg1: pypto.Tensor,
    arg2: pypto.Tensor,
    ...
) -> pypto.Tensor:
    ...

参数说明#

暂无参数配置。装饰器直接应用于函数。

返回值说明#

返回装饰后的可复用函数对象。

约束说明#

1. 只支持函数名调用，不支持方法调用

2. 参数：提供完整数量(无缺省值)严格按声明顺序传递(无乱序/关键词)

3. 不支持递归函数

与pypto.frontend.jit的区别#