Star0

Star0

Files	Last commit	Last update
examples	增加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	10 months ago
op_graph	add kirinx90 ops Co-authored-by: zengjuan<zengjuan2@huawei.com> # message auto-generated for no-merge-commit merge: !535 merge master into master add kirinx90 ops Created-by: zengjuan Commit-by: zengjuan Merged-by: cann-robot Description: ## 描述 add kirinx90 ops ## 关联的Issue 无 ## 测试黄区/蓝区门禁、算子冒烟 ## 文档更新无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!535	10 months ago
op_host	add kirinx90 ops Co-authored-by: zengjuan<zengjuan2@huawei.com> # message auto-generated for no-merge-commit merge: !535 merge master into master add kirinx90 ops Created-by: zengjuan Commit-by: zengjuan Merged-by: cann-robot Description: ## 描述 add kirinx90 ops ## 关联的Issue 无 ## 测试黄区/蓝区门禁、算子冒烟 ## 文档更新无 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!535	10 months ago
op_kernel	增加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	10 months ago
tests	增加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	10 months ago
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	10 months ago
README.md	公共资料同步以及修改部分描述 Co-authored-by: unknown<leilin2@huawei.com> # message auto-generated for no-merge-commit merge: !623 merge fix_doc_backup into master 公共资料同步以及修改部分描述 Created-by: gcw_CRjbEjwl Commit-by: unknown Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 1、修改docs下README中部分链接地址 2、修改910b Atlas A2的描述 3、提供910D数据类型 4、aicore_develop_guide.md的link失效 5、删除融合pass 6、perblock量化.png修复， 7、op_debug_prof.md优化描述 8、examples下单个算子readme修改 9、环境部署增加跳转指引 10、增加离线编译资料说明 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> 不涉及 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> 不涉及 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> README.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!623	10 months ago

StackBallQuery

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

功能说明

算子功能：Stack Ball Query 是KNN的替代方案，用于查找点p1指定半径范围内的所有点（在实现中设置了K的上限）。返回的点不一定是距离p1最近的点，而是半径范围内的前K个点。和Ball Query算子相比，Stack Ball Query将Ball Query算子的输入进行了堆叠，center_xyz和xyz的维度从3变成了2。优势：当半径范围内存在大量点时，该算法比KNN算法更快，同时保证了固定的区域尺度，使局部区域特征更通用。使用模型：该算子在PointNet++模型中被提出，在该模型及其衍生模型中使用。 Stack Ball Query具体操作如下：根据输入的center_xyz, 对同一个batch内的每一个点计算和xyz之间的距离。如果距离小于max_radius，保存xyz点的索引值。寻找到sample_num个满足要求的索引后，退出循环。输出保存的索引值。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
xyz	输入	2D Tensor , 经过堆叠的xyz 的坐标值	FLOAT16、FLOAT32	ND
center_xyz	输入	2D Tensor , 经过堆叠的center_xyz 的坐标值	FLOAT16、FLOAT32	ND
xyz_batch_cnt	输入	1D Tensor, 表示每个batch中的xyz点个数	INT32、INT64	ND
center_xyz_batch_cnt	输入	1D Tensor, 表示每个batch中的center_xyz点个数	INT32、INT64	ND
idx	输出	stack ball query后得到的索引值	INT32	ND
max_radius	输入属性	最大半径值	FLOAT	-
sample_num	输入属性	最大采样数	INT	-

约束说明

无。

调用说明

调用方式	样例代码	说明
aclnn接口	test_stack_ball_query	通过aclnnStackBallQuery接口方式调用StackBallQuery算子。