已关闭
[Requirement|需求建议]: 支持LeftShift算子TensorFlow框架插件 #2202
songkai111创建于  7月10日关闭于  7月10日
songkai111成员
7月10日 创建

一、背景信息 (必填)

ops-math 仓库需要支持 LeftShift 算子的 TensorFlow 框架适配。当前 PR 新增 LeftShift 算子的 TensorFlow 插件(math/left_shift/framework/left_shift_tf_plugin.cpp),通过 REGISTER_CUSTOM_OP 注册算子,完成 TensorFlow 框架下 LeftShift 算子的接入。

二、价值/作用 (必填)

使 ops-math 仓库具备在 TensorFlow 框架下解析和调度 LeftShift 算子的能力,支撑 TensorFlow 原生 tf.bitwise.left_shift 算子在昇腾平台上的运行。

三、设计方案 (必填)

3.1 使能方式

TensorFlow 框架(代码依据:FrameworkType(TENSORFLOW)),算子执行在 AI_CPU(代码依据:ImplyType(ImplyType::AI_CPU))。

3.2 总体设计

通过 REGISTER_CUSTOM_OP("LeftShift") 注册算子,使用 AutoMappingByOpFn 实现参数的自动映射解析,无需手动编写参数解析逻辑。

3.2.1 算子支持的数据类型

见后续算子实现补充。

3.2.2 host侧设计

插件层使用 AutoMappingByOpFn 自动映射参数,无需额外的 host 侧解析逻辑。

3.2.3 kernel侧设计

见后续算子 kernel 实现补充。

3.3 支持硬件

AI_CPU(代码依据:ImplyType(ImplyType::AI_CPU))。

3.4 算子约束限制

见后续算子实现补充。

💡 备注
关联 PR:https://gitcode.com/cann/ops-math/pull/3957

likedislike
Ssongkai111成员
7月10日 添加了label:requirement
songkai111成员
7月10日 评论:

/assign @songkai111

likedislike
CANN-robotCANN-robot成员
7月10日 将 songkai111 设为负责人
CANN-robotCANN-robot成员
7月10日 关闭了 issue
CANN-robotCANN-robot成员
7月10日 添加了label:resolved