Star
136
Fork
218
代码
介绍
代码
Issues
19
Pull Requests
31
流水线
Actions
项目讨论
Wiki
项目成员
129
分析
项目设置
Star
136
Fork
218
已关闭
RecSDK hybrid 适配 rec_cust_ops 和 fbgemm_ascend
#1350
zengxiong
创建于
8月29日
关闭于
9月2日
Z
zengxiong
成员
8月29日
创建
8月29日
创建
背景
training/torch_rec_v1/hybrid_torchrec 当前对查表算子与自定义算子的调用方式,需要适配两项上游改动:
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 注册。
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 包安装后仍能正确加载算子。
关联
关联 MR:Ascend/RecSDK!3061(mxrec 查表命名空间迁移 + presmoke 改用 rec_cust_ops 包编译安装)
MR 链接:
https://gitcode.com/Ascend/RecSDK/merge_requests/3061
保存
取消
taoqiuyu1998
成员
8月29日
评论:
8月29日
评论:
/label add triaged
保存
取消
ascend-robot
成员
8月29日
添加了label:
triaged
8月29日
添加了label:
triaged
Z
zengxiong
成员
8月29日
关联了pull request:style: apply clang-format to torch_library (no functional change)
8月29日
关联了pull request:style: apply clang-format to torch_library (no functional change)
Z
zengxiong
成员
9月2日
issue状态由 TODO 改变为 DONE
9月2日
issue状态由 TODO 改变为 DONE
Z
zengxiong
成员
9月2日
关闭了 issue
9月2日
关闭了 issue
ascend-robot
成员
9月2日
添加了label:
resolved
9月2日
添加了label:
resolved
taoqiuyu1998
成员
9月8日
关联了看板:MindSDK版本issue看板
9月8日
关联了看板:MindSDK版本issue看板
背景
training/torch_rec_v1/hybrid_torchrec 当前对查表算子与自定义算子的调用方式,需要适配两项上游改动:
任务
关联