已关闭
RecSDK hybrid 适配 rec_cust_ops 和 fbgemm_ascend #1350
zengxiong创建于  8月29日关闭于  9月2日
zengxiong成员
8月29日 创建

背景

training/torch_rec_v1/hybrid_torchrec 当前对查表算子与自定义算子的调用方式,需要适配两项上游改动:

  1. rec_cust_ops 包编译安装:算子 OPP 不再由 presmoke/run.sh 逐算子独立编译注册,而是统一由 rec_cust_ops python 包(cust_op/setup.py + build_whl.sh)编译安装。import rec_cust_ops 即自动设置 ASCEND_CUSTOM_OPP_PATH 并加载对应芯片 host library,完成 OPP 注册。
  2. fbgemm_ascend recops 命名空间迁移:split_embedding 查表算子已从 fbgemm/mxrec 老命名(如 aclnnBackwardCodegenAdagradUnweightedExact)迁移到 recops 命名空间(如 aclnnRecopsBackwardCodegenAdagradUnweightedExact),并在 torch_plugin 侧新增 TORCH_LIBRARY_FRAGMENT(mxrec, m) 注册。

任务

  • hybrid_torchrec 中 import 与调用路径切换为 rec_cust_ops 包方式(import rec_cust_ops,不再依赖独立算子 OPP 路径注册)。
  • 同步适配 fbgemm_ascend recops 命名空间下的算子符号与调用约定。
  • 确保 presmoke / torchrec 用例在 rec_cust_ops 包安装后仍能正确加载算子。

关联

likedislike
taoqiuyu1998
taoqiuyu1998成员
8月29日 评论:

/label add triaged

likedislike
ascend-robotascend-robot成员
8月29日 添加了label:triaged
Zzengxiong成员
8月29日 关联了pull request:style: apply clang-format to torch_library (no functional change)
Zzengxiong成员
9月2日 issue状态由 TODO 改变为 DONE
Zzengxiong成员
9月2日 关闭了 issue
ascend-robotascend-robot成员
9月2日 添加了label:resolved
taoqiuyu1998taoqiuyu1998成员
9月8日 关联了看板:MindSDK版本issue看板