master

cann-robotAsStridedToContiguous

471c4a5c创建于 4 天前历史提交

文件	最后提交记录	最后更新时间
op_api	AsStridedToContiguous Co-authored-by: qq_45996812<liaohuming@huawei.com> # message auto-generated for no-merge-commit merge: !2986 merge master into master AsStridedToContiguous Created-by: liaohuming Commit-by: qq_45996812 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 在contiguous调as_strided前将输入shape减去offset。 ``` const aclTensor* AsStridedToContiguous(const aclTensor* x, aclOpExecutor* executor) { auto sizeV = op::ToShapeVector(x->GetViewShape()); auto size = executor->ConvertToTensor(sizeV.data(), sizeV.size(), TYPE_INT64); auto strides = x->GetViewStrides(); auto stride = executor->ConvertToTensor(strides.data(), strides.size(), TYPE_INT64); int64_t offset[1] = {0}; auto storageOffset = executor->ConvertToTensor(offset, 1, TYPE_INT64); auto out = executor->AllocTensor(x->GetViewShape(), x->GetDataType()); **op::Shape newStorageShape{1}; int64_t actualShapeSize = 1; auto tempStorageShape = x->GetStorageShape(); for (uint64_t i= 0; i < tempStorageShape.GetDimNum(); i++) { actualShapeSize *= tempStorageShape.GetDim(i); } actualShapeSize -= x->GetViewOffset(); newStorageShape.SetDim(0, actualShapeSize); auto xView = executor->CreateView(x, newStorageShape, x->GetViewOffset()); xView->SetViewShape(x->GetViewShape()); xView->SetViewStrides(x->GetViewStrides());** return AsStrided(**xView**, out, size, stride, storageOffset, executor); } ``` ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> https://gitcode.com/cann/ops-math/issues/1702 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!2986	4 天前
CMakeLists.txt	增加A2 A3新算子支持 Co-authored-by: slx2008<shaolixin@huawei.com> # message auto-generated for no-merge-commit merge: !448 merge master into master 增加A2 A3新算子支持 Created-by: songkai111 Commit-by: slx2008 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!448	5 个月前