op plugin support non-contiguous tensor input · Pull Request !90299 · MindSpore/mindspore

1. op plugin KernelInputInfo store tensor storage info
---
<!--  Thanks for sending a pull request!  Here are some tips for you:

1) If this is your first time, please read our contributor guidelines: https://gitee.com/mindspore/mindspore/blob/master/CONTRIBUTING.md

2) If you want to contribute your code but don't know who will review and merge, please add label `mindspore-assistant` to the pull request, we will find and do it as soon as possible.
-->

**What type of PR is this?**

/kind task

**What does this PR do / why do we need it**:

### 背景描述

op plugin流程传递device address, shape, dtype等信息到ms_op_plugin库后生成at::Tensor，传给torch的at接口进行计算。

当前因为缺失tensor storage信息，op plugin仅支持生成内存连续的tensor，不支持非连续tensor

### 设计思路

将tensor storage info（strides, offset）信息通过KernelInputInfo传递给op plugin算子，调用torch相关接口构造非连续tensor，提供给后续计算。

### 与其他模块的相关性描述

仅涉及op plugin流程。

### 其他信息（测试验证设计等）

在mock算子bitwise_and_mock.cc中使用tensor storage信息处理非连续输入，增加st用例tests/st/ops/op_plugin/test_op_plugin.py::test_noncontiguous_input_op 对非连续输入进行测试

**Which issue(s) this PR fixes**:

Fixes #ID5C9Z

---

**Test Plan and Test result：What scenarios were tested, and what were the verification results（Function, performance, reliability, etc.）**：

![输入图片说明](https://foruda.gitee.com/images/1762428061824965215/ed6b9d8f_10906943.png "屏幕截图")

**Self-checklist**:（**请自检，在[ ]内打上x，我们将检视你的完成情况，否则会导致pr无法合入**）

+ - [x] **设计**：PR对应的方案是否已经经过Maintainer评审，方案检视意见是否均已答复并完成方案修改
+ - [x] **测试**：PR中的代码是否已有UT/ST测试用例进行充分的覆盖，新增测试用例是否随本PR一并上库或已经上库
+ - [x] **验证**：PR描述信息中是否已包含对该PR对应的Feature、Refactor、Bugfix的预期目标达成情况的详细验证结果描述
+ - [ ] **接口**：是否涉及对外接口变更，相应变更已得到接口评审组织的通过，API对应的注释信息已经刷新正确
+ - [ ] **文档**：是否涉及官网文档修改，如果涉及请及时提交资料到Doc仓

GVP MindSpore/mindspore
关闭

发现个问题

GVPMindSpore/mindspore关闭

搜索帮助

GVP MindSpore/mindspore
关闭