ops-math/math/pdist/docs · focusforce/ops-math

cann-robot[CANNBOT] Pdist新增Ascend950 pypto实现

Files	Last commit	Last update
aclnnPdist.md	[CANNBOT] Pdist新增Ascend950 pypto实现 Co-authored-by: SimonZzz<zhouquan79@huawei.com> # message auto-generated for no-merge-commit merge: !5438 merge pdist_pypto into master [CANNBOT] Pdist新增Ascend950 pypto实现 Created-by: SimonZzz Commit-by: SimonZzz Merged-by: cann-robot Description: ## 描述 pdist 算子新增 Ascend 950PR/950DT 系列（reg-base，DAV_3510/DAV_5102）PyPTO 内核实现，补齐 host 侧 tiling、GE 注册/推导到测试资产的全链路改动： - PyPTO kernel（op_kernel/pdist.py）：m-lane row-major 布局（零转置，消除 v1 pair-lane 转置布局下 vscatter 导致的 UB bank 串行化，fp16 [8192,512] 由 963ms 优化至约 20.6ms）；M_BLK=128 分块 ping-pong 部分和累加（含 p=1/inf 分支的 NaN 计数）；按 PBranch 编译期折叠的无分支内层 m 循环；提供 fp32/fp16 双 dtype 实例（fp16 以 pl.cast 展宽至 fp32 计算、出口 CAST_RINT 收窄） - host tiling（op_host/arch35/pdist_tiling.cpp/.h）：p 值映射 0-5 的 PBranch 分支并以 GET_TPL_TILING_KEY 编码 TilingKey；闭式 PdistFloorSum 计算 64 对任务块数 B，按核连续块域切分，MIX 发射方式计算 blockDim；tiling 数据含 N/M/B/p_val/inv_p 字段 - GE 注册与推导（op_host/pdist_def.cpp、pdist_infershape.cpp）：ascend950 AICore 配置及 DynamicShapeSupportFlag 等动态开关；输出 y=[N(N-1)/2]，dtype 与输入一致 - 测试资产*：examples/test_aclnn_pdist.cpp 重构为带 golden 容差比对的用例（fp32 覆盖 p=0/0.5/1/2/3/inf、大 M 多块场景及 fp16 用例）；新增 GE IR RunGraph 用例 test_geir_pdist.cpp；tests/assets/golden.py 提供上三角行优先输出语义的参考实现与 TTK TestSpec（PdistTestSpec/AclnnPdistTestSpec/AclnnPdistForwardTestSpec）注册及 fp16/fp32 容差配置 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4169 ## 测试 - kernel 移植自 Stage-4 验证通过的 PyPTO-Pro v2 交付（verifier PASS 15/15、16/16 用例，附 Stage 5 性能证据） - eager 模式示例 fp32（p=0/0.5/1/2/3/inf、大 M 多块）与 fp16 用例均与 golden 容差比对通过 - 本地 pre-commit 全量通过（clang-format / ruff / codespell / OAT 合规），CI CodeCheck 门禁通过；PreSmoke 示例已去除 securec 依赖修复链接失败，随本提交重新验证 ## 文档更新更新 math/pdist/README.md、docs/aclnnPdist.md、docs/aclnnPdistForward.md（实现方案、参数与产品支持说明） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5438	16 days ago
aclnnPdistForward.md	[CANNBOT] Pdist新增Ascend950 pypto实现 Co-authored-by: SimonZzz<zhouquan79@huawei.com> # message auto-generated for no-merge-commit merge: !5438 merge pdist_pypto into master [CANNBOT] Pdist新增Ascend950 pypto实现 Created-by: SimonZzz Commit-by: SimonZzz Merged-by: cann-robot Description: ## 描述 pdist 算子新增 Ascend 950PR/950DT 系列（reg-base，DAV_3510/DAV_5102）PyPTO 内核实现，补齐 host 侧 tiling、GE 注册/推导到测试资产的全链路改动： - PyPTO kernel（op_kernel/pdist.py）：m-lane row-major 布局（零转置，消除 v1 pair-lane 转置布局下 vscatter 导致的 UB bank 串行化，fp16 [8192,512] 由 963ms 优化至约 20.6ms）；M_BLK=128 分块 ping-pong 部分和累加（含 p=1/inf 分支的 NaN 计数）；按 PBranch 编译期折叠的无分支内层 m 循环；提供 fp32/fp16 双 dtype 实例（fp16 以 pl.cast 展宽至 fp32 计算、出口 CAST_RINT 收窄） - host tiling（op_host/arch35/pdist_tiling.cpp/.h）：p 值映射 0-5 的 PBranch 分支并以 GET_TPL_TILING_KEY 编码 TilingKey；闭式 PdistFloorSum 计算 64 对任务块数 B，按核连续块域切分，MIX 发射方式计算 blockDim；tiling 数据含 N/M/B/p_val/inv_p 字段 - GE 注册与推导（op_host/pdist_def.cpp、pdist_infershape.cpp）：ascend950 AICore 配置及 DynamicShapeSupportFlag 等动态开关；输出 y=[N(N-1)/2]，dtype 与输入一致 - 测试资产*：examples/test_aclnn_pdist.cpp 重构为带 golden 容差比对的用例（fp32 覆盖 p=0/0.5/1/2/3/inf、大 M 多块场景及 fp16 用例）；新增 GE IR RunGraph 用例 test_geir_pdist.cpp；tests/assets/golden.py 提供上三角行优先输出语义的参考实现与 TTK TestSpec（PdistTestSpec/AclnnPdistTestSpec/AclnnPdistForwardTestSpec）注册及 fp16/fp32 容差配置 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4169 ## 测试 - kernel 移植自 Stage-4 验证通过的 PyPTO-Pro v2 交付（verifier PASS 15/15、16/16 用例，附 Stage 5 性能证据） - eager 模式示例 fp32（p=0/0.5/1/2/3/inf、大 M 多块）与 fp16 用例均与 golden 容差比对通过 - 本地 pre-commit 全量通过（clang-format / ruff / codespell / OAT 合规），CI CodeCheck 门禁通过；PreSmoke 示例已去除 securec 依赖修复链接失败，随本提交重新验证 ## 文档更新更新 math/pdist/README.md、docs/aclnnPdist.md、docs/aclnnPdistForward.md（实现方案、参数与产品支持说明） ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!5438	16 days ago
aclnnPdistForward_en.md	master翻译合入 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !6751 merge master into master master翻译合入 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述合入math仓翻译的内容 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4170 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 docs/context，aclnn等商发文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6751	17 days ago
aclnnPdist_en.md	master翻译合入 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !6751 merge master into master master翻译合入 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述合入math仓翻译的内容 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/4170 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 docs/context，aclnn等商发文档 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!6751	17 days ago