已关闭
[feature] hybrid_torchrec: 不 import fbgemm_ascend,NPU lib 加载降级处理 #1339
zengxiong创建于  14 天前关闭于  2 天前
zengxiong成员
14 天前 创建

背景 / 设计意图

这是有意为之的特性改动(见 PR #3040):

  • fbgemm_ascend 1.2.0 在 torch.ops.fbgemm 下注册了与 cust_op A2 冲突的 split_embedding_codegen_lookup_adagrad_function(A5 实现)。两者同时加载会触发 C++ 重复 TORCH_LIBRARY 定义;仅加载 fbgemm_ascend 则在 A2 上得到错误的(A5)结果。
  • 因此刻意改为:不再 import fbgemm_ascend,统一通过 libfbgemm_npu_api.so(cust_op A2,以 TORCH_LIBRARY_FRAGMENT 挂载到 fbgemm_gpu)加载 A2 正确的 NPU 算子;
  • libfbgemm_npu_api.so 加载失败降级为非致命(无 libtorch_npu.so 的纯 CPU 上下文仍可正常 import)。

关联

  • PR: #3040 (fix/hybrid_torchrec_no_fbgemm_ascend_import_20260819)
likedislike
xiangjie10成员
14 天前 评论:

👋 您好,感谢向 RecSDK 提交 Issue!
🎉 我们已收到您的反馈,感谢你对开源社区的支持!

📅 处理时效 维护团队将在工作日 24 小时内查看并回复您的问题。
🔍 自助排查(推荐优先查看) 在等待回复期间,您可以先查阅仓库README以及历史 Issue 中相似问题的解决方案,多数问题可快速解决。
💡 为了更快定位问题,请您确保 Issue 包含:

  • 清晰的问题描述
  • 可复现的操作步骤
  • 相关日志、截图或环境信息
    我们会尽快跟进,感谢您的理解与配合!
likedislike
xiangjie10成员
14 天前 评论:

/label add triaged

likedislike
ascend-robotascend-robot成员
14 天前 添加了label:bug
ascend-robotascend-robot成员
14 天前 添加了label:triaged
Zzengxiong成员
14 天前 修改标题为 “[feature] hybrid_torchrec: 不 import fbgemm_ascend,NPU lib 加载降级处理”,原标题为“[bug] hybrid_torchrec 不应 import fbgemm_ascend,NPU lib 加载应降级而非致命”
Zzengxiong成员
14 天前 修改了issue 的描述
ascend-robotascend-robot成员
14 天前 添加了label:feature
Zzengxiong成员
14 天前 issue类型由 Bug-Report 改变为 需求
Zzengxiong成员
2 天前 issue状态由 TODO 改变为 DONE
Zzengxiong成员
2 天前 关闭了 issue
ascend-robotascend-robot成员
2 天前 添加了label:resolved