尧图网站建设 
PyPTO Kernel 标量最小值运算指南Python 内置 min 与 pypto_pro.language.min 的自动转换机制【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto导读PyPTOParallel Tensor/Tile Operation 编程范式为 Kernel 编程提供了一套贴近 Python 语法的算子表达方式。本指南围绕pypto_pro.language.min展开讲解如何在 Kernel 代码中直接使用 Python 内置min(lhs, rhs)完成标量最小值计算编译器会自动将其转换为pypto_pro.language.min涵盖函数原型、参数与返回值约定、约束条件、调用示例并结合仓库源码剖析其底层转换与常量折叠实现帮助读者在循环边界裁剪、索引计算、因果掩码等典型场景中正确、高效地使用该标量运算。产品支持情况pypto_pro.language.min在不同昇腾产品上的支持情况如下与 min.md 记录一致产品支持情况Ascend 950PR / Ascend 950DT支持Atlas A3 训练系列产品 / Atlas A3 推理系列产品不支持Atlas A2 训练系列产品 / Atlas A2 推理系列产品不支持在目标设备上编译 Kernel 前请先确认当前产品型号是否在支持列表内。功能说明内置 min 的自动转换在 PyPTO Kernel 代码中可以直接使用 Python 内置函数min(lhs, rhs)求两个标量的较小值。编译时解析器会自动把对内置min的调用转换为pypto_pro.language.min(lhs, rhs)因此两种写法完全等价无需显式区分。该自动转换在解析器层通过内置函数到 pl.* 算子的映射表实现见 python/pypto_pro/language/parser/_call_parser.py# Builtin function names that map to pl.* ops (syntax sugar). _BUILTIN_TO_OP: dict[str, str] { min: min, max: max, }当 Kernel 内出现裸函数名min(...)调用时调用解析器会在 python/pypto_pro/language/parser/_call_parser.py 中查询该映射表将内置min路由到pl.min即pypto_pro.language.min对应的算子解析函数# Builtin min/max - route to pl.min/pl.max (scalar_ops.py) if func_name in _BUILTIN_TO_OP: op_func _OP_REGISTRY.get(_BUILTIN_TO_OP[func_name]) if op_func is not None: return op_func(self, call)因此min(a, b)与pl.min(a, b)在编译链路中殊途同归最终都会落到pypto_pro.language.min这一标量运算上。min与max均已从 python/pypto_pro/language/init.py 导出可直接通过pl.min访问。函数原型# 以下两种写法等价 result min(lhs, rhs) result pypto_pro.language.min(lhs, rhs)在源码层面pypto_pro.language.min的声明位于 python/pypto_pro/language/_api.py其 docstring 明确指出该操作仅用于标量Scalar-only operation服务于循环边界计算等场景并提示Tile 逐元素最小值请使用pl.minimum_api_decl def min(lhs: Scalar, rhs: Scalar) - Scalar: Return the minimum of two scalars. Scalar-only operation for loop-bound calculations etc. For tile element-wise minimum, use pl.minimum. 参数说明参数输入/输出说明lhs输入左操作数Python int、Python float 或 Kernel 内整型或浮点型标量表达式rhs输入右操作数Python int、Python float 或 Kernel 内整型或浮点型标量表达式两个操作数既可以是宿主侧host的 Python 常量也可以是 Kernel 内的标量表达式——例如由循环变量、维度元数据派生的表达式m_dim - i、n_dim - j等。返回值类型与操作数类型一致。约束说明使用min时需遵守以下约束否则会在编译期报错仅用于标量min面向标量运算典型用途是循环边界、索引计算等场景它不适用于 Tile 的逐元素最小值计算。Tile 逐元素取最小值若需要对 Tile 中每个元素取最小值应使用pypto_pro.language.minimum。在 python/pypto_pro/language/_api.py 中minimum是面向 Tile 的重载算子支持 tile-tile、tile-scalar 逐元素模式以及带dim参数的按维度 min 归约模式dim0行方向、dim1列方向。不支持多参数min仅接受恰好 2 个参数min(a, b, c)这类多参数写法不支持需要嵌套调用如min(min(a, b), c)。同类别约束两个操作数须同为整型或同为浮点型混合 int/float 会报错如需混合使用请先做显式类型转换使两侧类型一致。返回值说明返回lhs和rhs中较小的标量值。当两个操作数相等时返回该相等值。返回值类型保持与操作数相同的类别整型或浮点型可直接参与后续标量运算或作为循环边界、切片尺寸使用。调用示例以下示例来自 min.md展示了min在循环边界裁剪、切片尺寸计算与因果掩码等场景中的典型用法import pypto_pro.language as pl pl.jit() def example_kernel(...): m_size min(m_dim - i, 64) n_size pl.min(n_dim - j, 128) causal_kv_tiles min(qi 1, skv_tiles) bottom min(a, 0)其中m_size min(m_dim - i, 64)与n_size pl.min(n_dim - j, 128)分别演示了裸内置min与pl.min两种等价写法二者的作用是相同的把分块循环的边界裁剪到不超过 64 / 128避免越界访问causal_kv_tiles min(qi 1, skv_tiles)则用于注意力机制中限制因果掩码的 KV 分块数量。在仓库测试中pl.min被广泛用于计算 Tile 的合法尺寸valid size例如 python/tests/st/pypto_pro/frontend/control_flow/test_for.py 中的模式valid_r pl.min(m - i, TILE_M) valid_c pl.min(n - j, TILE_N)这里用pl.min取剩余行数与Tile 行尺寸的较小者确保最后一个不完整 Tile 的边界计算正确是min在 Kernel 循环中的高频用法。同一模式也出现在 python/tests/st/pypto_pro/frontend/control_flow/test_break.py 与 python/tests/st/pypto_pro/frontend/control_flow/test_continue.py 中可作为参考用例。底层实现常量折叠与编译优化除了把内置min路由到pl.min之外PyPTO 的表达式解析器还支持对min/max的编译期常量折叠当两个操作数在编译期都能求出常量值时会直接计算出最小值并生成常量避免在 Kernel 中引入运行时指令。该逻辑位于 python/pypto_pro/language/parser/_expression_parser.pybinary_ops { add: lambda: left_value right_value, ... min: lambda: min(left_value, right_value), max: lambda: max(left_value, right_value), }从源码结构可以推断min与max、minimum与maximum成对实现见 python/pypto_pro/language/_api.py 中max与maximum的声明对称地覆盖标量与 Tile 两个层面的极值运算需求min/max的标量实现位于scalar_ops.pypl.*标量算子模块这与调用解析器中的注释route to pl.min/pl.max (scalar_ops.py)相印证。总结在 Kernel 中min(lhs, rhs)与pypto_pro.language.min(lhs, rhs)完全等价编译器通过 python/pypto_pro/language/parser/_call_parser.py 中的_BUILTIN_TO_OP映射自动完成转换。min仅接受恰好 2 个同类别同为整型或同为浮点型的标量操作数适合循环边界裁剪、索引计算、切片尺寸与因果掩码等场景。Tile 的逐元素最小值计算应使用pypto_pro.language.minimum二者职责不同不可混用。在使用前请确认目标产品Ascend 950PR / Ascend 950DT 支持Atlas A2/A3 系列暂不支持。【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考 PyPTO Kernel 标量最小值运算指南：Python 内置 min 与 pypto_pro.language.min 的自动转换机制
PyPTO Kernel 标量最小值运算指南Python 内置 min 与 pypto_pro.language.min 的自动转换机制【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto导读PyPTOParallel Tensor/Tile Operation 编程范式为 Kernel 编程提供了一套贴近 Python 语法的算子表达方式。本指南围绕pypto_pro.language.min展开讲解如何在 Kernel 代码中直接使用 Python 内置min(lhs, rhs)完成标量最小值计算编译器会自动将其转换为pypto_pro.language.min涵盖函数原型、参数与返回值约定、约束条件、调用示例并结合仓库源码剖析其底层转换与常量折叠实现帮助读者在循环边界裁剪、索引计算、因果掩码等典型场景中正确、高效地使用该标量运算。产品支持情况pypto_pro.language.min在不同昇腾产品上的支持情况如下与 min.md 记录一致产品支持情况Ascend 950PR / Ascend 950DT支持Atlas A3 训练系列产品 / Atlas A3 推理系列产品不支持Atlas A2 训练系列产品 / Atlas A2 推理系列产品不支持在目标设备上编译 Kernel 前请先确认当前产品型号是否在支持列表内。功能说明内置 min 的自动转换在 PyPTO Kernel 代码中可以直接使用 Python 内置函数min(lhs, rhs)求两个标量的较小值。编译时解析器会自动把对内置min的调用转换为pypto_pro.language.min(lhs, rhs)因此两种写法完全等价无需显式区分。该自动转换在解析器层通过内置函数到 pl.* 算子的映射表实现见 python/pypto_pro/language/parser/_call_parser.py# Builtin function names that map to pl.* ops (syntax sugar). _BUILTIN_TO_OP: dict[str, str] { min: min, max: max, }当 Kernel 内出现裸函数名min(...)调用时调用解析器会在 python/pypto_pro/language/parser/_call_parser.py 中查询该映射表将内置min路由到pl.min即pypto_pro.language.min对应的算子解析函数# Builtin min/max - route to pl.min/pl.max (scalar_ops.py) if func_name in _BUILTIN_TO_OP: op_func _OP_REGISTRY.get(_BUILTIN_TO_OP[func_name]) if op_func is not None: return op_func(self, call)因此min(a, b)与pl.min(a, b)在编译链路中殊途同归最终都会落到pypto_pro.language.min这一标量运算上。min与max均已从 python/pypto_pro/language/init.py 导出可直接通过pl.min访问。函数原型# 以下两种写法等价 result min(lhs, rhs) result pypto_pro.language.min(lhs, rhs)在源码层面pypto_pro.language.min的声明位于 python/pypto_pro/language/_api.py其 docstring 明确指出该操作仅用于标量Scalar-only operation服务于循环边界计算等场景并提示Tile 逐元素最小值请使用pl.minimum_api_decl def min(lhs: Scalar, rhs: Scalar) - Scalar: Return the minimum of two scalars. Scalar-only operation for loop-bound calculations etc. For tile element-wise minimum, use pl.minimum. 参数说明参数输入/输出说明lhs输入左操作数Python int、Python float 或 Kernel 内整型或浮点型标量表达式rhs输入右操作数Python int、Python float 或 Kernel 内整型或浮点型标量表达式两个操作数既可以是宿主侧host的 Python 常量也可以是 Kernel 内的标量表达式——例如由循环变量、维度元数据派生的表达式m_dim - i、n_dim - j等。返回值类型与操作数类型一致。约束说明使用min时需遵守以下约束否则会在编译期报错仅用于标量min面向标量运算典型用途是循环边界、索引计算等场景它不适用于 Tile 的逐元素最小值计算。Tile 逐元素取最小值若需要对 Tile 中每个元素取最小值应使用pypto_pro.language.minimum。在 python/pypto_pro/language/_api.py 中minimum是面向 Tile 的重载算子支持 tile-tile、tile-scalar 逐元素模式以及带dim参数的按维度 min 归约模式dim0行方向、dim1列方向。不支持多参数min仅接受恰好 2 个参数min(a, b, c)这类多参数写法不支持需要嵌套调用如min(min(a, b), c)。同类别约束两个操作数须同为整型或同为浮点型混合 int/float 会报错如需混合使用请先做显式类型转换使两侧类型一致。返回值说明返回lhs和rhs中较小的标量值。当两个操作数相等时返回该相等值。返回值类型保持与操作数相同的类别整型或浮点型可直接参与后续标量运算或作为循环边界、切片尺寸使用。调用示例以下示例来自 min.md展示了min在循环边界裁剪、切片尺寸计算与因果掩码等场景中的典型用法import pypto_pro.language as pl pl.jit() def example_kernel(...): m_size min(m_dim - i, 64) n_size pl.min(n_dim - j, 128) causal_kv_tiles min(qi 1, skv_tiles) bottom min(a, 0)其中m_size min(m_dim - i, 64)与n_size pl.min(n_dim - j, 128)分别演示了裸内置min与pl.min两种等价写法二者的作用是相同的把分块循环的边界裁剪到不超过 64 / 128避免越界访问causal_kv_tiles min(qi 1, skv_tiles)则用于注意力机制中限制因果掩码的 KV 分块数量。在仓库测试中pl.min被广泛用于计算 Tile 的合法尺寸valid size例如 python/tests/st/pypto_pro/frontend/control_flow/test_for.py 中的模式valid_r pl.min(m - i, TILE_M) valid_c pl.min(n - j, TILE_N)这里用pl.min取剩余行数与Tile 行尺寸的较小者确保最后一个不完整 Tile 的边界计算正确是min在 Kernel 循环中的高频用法。同一模式也出现在 python/tests/st/pypto_pro/frontend/control_flow/test_break.py 与 python/tests/st/pypto_pro/frontend/control_flow/test_continue.py 中可作为参考用例。底层实现常量折叠与编译优化除了把内置min路由到pl.min之外PyPTO 的表达式解析器还支持对min/max的编译期常量折叠当两个操作数在编译期都能求出常量值时会直接计算出最小值并生成常量避免在 Kernel 中引入运行时指令。该逻辑位于 python/pypto_pro/language/parser/_expression_parser.pybinary_ops { add: lambda: left_value right_value, ... min: lambda: min(left_value, right_value), max: lambda: max(left_value, right_value), }从源码结构可以推断min与max、minimum与maximum成对实现见 python/pypto_pro/language/_api.py 中max与maximum的声明对称地覆盖标量与 Tile 两个层面的极值运算需求min/max的标量实现位于scalar_ops.pypl.*标量算子模块这与调用解析器中的注释route to pl.min/pl.max (scalar_ops.py)相印证。总结在 Kernel 中min(lhs, rhs)与pypto_pro.language.min(lhs, rhs)完全等价编译器通过 python/pypto_pro/language/parser/_call_parser.py 中的_BUILTIN_TO_OP映射自动完成转换。min仅接受恰好 2 个同类别同为整型或同为浮点型的标量操作数适合循环边界裁剪、索引计算、切片尺寸与因果掩码等场景。Tile 的逐元素最小值计算应使用pypto_pro.language.minimum二者职责不同不可混用。在使用前请确认目标产品Ascend 950PR / Ascend 950DT 支持Atlas A2/A3 系列暂不支持。【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考