Skip to content

fix: add PIR operator mappers for linear_v2, group_norm, layer_norm#1652

Open
PlumBlossomMaid wants to merge 2 commits into
PaddlePaddle:developfrom
PlumBlossomMaid:fix/pir-op-mappers-v2
Open

fix: add PIR operator mappers for linear_v2, group_norm, layer_norm#1652
PlumBlossomMaid wants to merge 2 commits into
PaddlePaddle:developfrom
PlumBlossomMaid:fix/pir-op-mappers-v2

Conversation

@PlumBlossomMaid

@PlumBlossomMaid PlumBlossomMaid commented Jun 24, 2026

Copy link
Copy Markdown

修复内容

  1. linear_v2: 新增 PIR mapper,支持 PIR 格式的 Linear 算子导出
  2. group_norm: 修复 PIR 模式下输入为 3D 的问题(Unsqueeze→处理→Squeeze)
  3. layer_norm: 修复 PIR 模式下 GetInput("Scale") 应使用位置索引的问题

代码审查修复

  • Opset7 + Opset17: PIR 模式下改用 get_tensor lambda 通过位置索引访问 Scale/Bias
  • group_norm: 删除死代码 fallback block
  • group_norm: InstanceNormalization dummy_scale 匹配输入 dtype(修复 FP32 硬编码)

其他

  • mappers_registry.h.in 注册 PIR mapper
  • 新增 test_auto_scan_linear.py CI 测试

Fix #1644

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[PIR] linear_v2 / group_norm / layer_norm 算子导出失败 — PIR 位置索引无命名映射

1 participant