已开启
[Feature]: 库上看护模型适配fbgemm_ascend/ops_rec特性 #1346
zhoucy创建于 8 天前
8 天前 添加了label:feature
8 天前 关联了pull request:[feat]库上看护模型适配fbgemm_ascend/ops_rec特性
8 天前 将 zhoucy998 设为负责人
taoqiuyu1998
8 天前 评论:
8 天前 评论:
/label add triaged


8 天前 添加了label:triaged
6 天前 关联了pull request:[feat]库上看护模型适配fbgemm_ascend/ops_rec特性
提交提案之前,请先检索仓库内是否已有相同的提案,如已有请在同一提案中进行讨论。
💻 需求背景、当前现状、期望实现的功能内容、具体的设计方案、以及测试方案
现状
旧方式痛点:需手动 load_library,需编译产出 .so,路径依赖 sysconfig,易出错。
目标
新方式优势:import 即自动完成 NPU 算子注册绑定,无需手动 load_library。
核心规则:模型中使用的 fbgemm 相关接口均改为
torch.ops.fbgemm调用。原来通过torch.ops.mxrec.<fbgemm域算子>调用的(如 gr_nv 中的torch.ops.mxrec.jagged_to_padded_dense、torch.ops.mxrec.dense_to_jagged),必须改为torch.ops.fbgemm.<同名算子>。替代方案
补充说明
欢迎加入社区,感谢您对社区的贡献 🎉!