catccos/examples/allgather_matmul · CANN/catccos

cann-robotdocs: 修正样例文档与构建配置，并增强 setup.sh 传参

文件	最后提交记录	最后更新时间
scripts	docs: 修正样例文档与构建配置，并增强 setup.sh 传参 Co-authored-by: dovahkiiin<haorunzhe@h-partners.com> # message auto-generated for no-merge-commit merge: !81 merge docs/readme-setup-build-config into master docs: 修正样例文档与构建配置，并增强 setup.sh 传参 Created-by: dovahkiiin Commit-by: dovahkiiin Merged-by: cann-robot Description: ## 变更描述 / Description - 修改文档中对必选参数device_list的描述方式 - 在编译运行examples目录下算子时取消了对dynamic tiling相关内容的编译 - 增强了环境配置脚本setup.sh中对soc_type参数的校验（调用方式无变动，拦截了添加-soc_type参数时传递错误型号的场景） ## 改动类型 / Change Type - [ ] Bug 修复 / Bug Fix - [ ] 新功能 / New Feature - [ ] 性能优化 / Performance - [ ] 代码重构 / Refactoring - [x] 文档更新 / Documentation - [ ] 测试相关 / Test - [ ] 其它 / Other ## 关联 Issue / Related Issues <!-- Closes #000 可自动关闭 / Closes #000 to auto-close --> - Closes [#4](https://gitcode.com/cann/catccos/issues/4) - References # ## 测试信息 / Testing <!-- 简要测试说明或关键结果 / Brief test description or key results --> - [x] 单元测试通过 / UT passed - [ ] 集成测试通过 / ST passed - [x] 人工验证通过 / Manual verified ## 检查清单 / Checklist - [x] 代码符合规范 / Code follows style guide - [ ] 测试添加并通过 / Tests added and passed - [x] 文档已更新 / Docs updated if needed - [x] 无硬编码敏感信息 / No secrets hardcoded - [x] 提交信息符合规范 / Commit message follows convention See merge request: cann/catccos!81	1 个月前
testcases	init	4 个月前
CMakeLists.txt	feat: 增加allgather_matmul算子用customClass接入pytorch的示例和对应的skil。 Co-authored-by: soyabean<sunyanbin8@huawei.com> Co-authored-by: i_soyabean<i_soyabean@noreply.gitcode.com> # message auto-generated for no-merge-commit merge: !16 merge feat/add-agmm-py into master feat: add agmm pytorch support example Created-by: i_soyabean Commit-by: soyabean;i_soyabean Merged-by: cann-robot Description: feat: 增加allgather_matmul算子接入pytorch的示例。 See merge request: cann/catccos!16	2 个月前
README.md	docs: 修正样例文档与构建配置，并增强 setup.sh 传参 Co-authored-by: dovahkiiin<haorunzhe@h-partners.com> # message auto-generated for no-merge-commit merge: !81 merge docs/readme-setup-build-config into master docs: 修正样例文档与构建配置，并增强 setup.sh 传参 Created-by: dovahkiiin Commit-by: dovahkiiin Merged-by: cann-robot Description: ## 变更描述 / Description - 修改文档中对必选参数device_list的描述方式 - 在编译运行examples目录下算子时取消了对dynamic tiling相关内容的编译 - 增强了环境配置脚本setup.sh中对soc_type参数的校验（调用方式无变动，拦截了添加-soc_type参数时传递错误型号的场景） ## 改动类型 / Change Type - [ ] Bug 修复 / Bug Fix - [ ] 新功能 / New Feature - [ ] 性能优化 / Performance - [ ] 代码重构 / Refactoring - [x] 文档更新 / Documentation - [ ] 测试相关 / Test - [ ] 其它 / Other ## 关联 Issue / Related Issues <!-- Closes #000 可自动关闭 / Closes #000 to auto-close --> - Closes [#4](https://gitcode.com/cann/catccos/issues/4) - References # ## 测试信息 / Testing <!-- 简要测试说明或关键结果 / Brief test description or key results --> - [x] 单元测试通过 / UT passed - [ ] 集成测试通过 / ST passed - [x] 人工验证通过 / Manual verified ## 检查清单 / Checklist - [x] 代码符合规范 / Code follows style guide - [ ] 测试添加并通过 / Tests added and passed - [x] 文档已更新 / Docs updated if needed - [x] 无硬编码敏感信息 / No secrets hardcoded - [x] 提交信息符合规范 / Commit message follows convention See merge request: cann/catccos!81	1 个月前
allgather_matmul_device.h	feat: 修改AGMM pytorch接入方式，从customClass改为TORCH_LIBRARY，并更新对应skill Co-authored-by: soyabean<sunyanbin8@huawei.com> # message auto-generated for no-merge-commit merge: !51 merge feat/agmm-torch-library-registration into master feat: 修改AGMM pytorch接入方式，从customClass改为TORCH_LIBRARY，并更新对应skill Created-by: i_soyabean Commit-by: soyabean Merged-by: cann-robot Description: ## 变更描述 / Description <!-- 本 PR 做了什么，为什么需要 / What does this PR do and why --> AGMM算子之前通过PyTorch custom class方式接入，算子无法入图。现采用TORCH_LIBRARY注册方式注册为PyTorch自定义算子，可以通过`torch.ops.catccos.*`调用算子。 - 新增AGMM Torch Library 接入路径。 - 删除了原来的通过custom class的接入方式。 - 更新了对应的`torch-binding` SKILL。 ## 改动类型 / Change Type - [ ] Bug 修复 / Bug Fix - [x] 新功能 / New Feature - [ ] 性能优化 / Performance - [ ] 代码重构 / Refactoring - [ ] 文档更新 / Documentation - [ ] 测试相关 / Test - [ ] 其它 / Other ## 关联 Issue / Related Issues <!-- Closes #000 可自动关闭 / Closes #000 to auto-close --> - Closes # - References # ## 测试信息 / Testing <!-- 简要测试说明或关键结果 / Brief test description or key results --> - [x] 单元测试通过 / UT passed - [x] 集成测试通过 / ST passed - [x] 人工验证通过 / Manual verified ## 检查清单 / Checklist - [ ] 代码符合规范 / Code follows style guide - [ ] 测试添加并通过 / Tests added and passed - [ ] 文档已更新 / Docs updated if needed - [ ] 无硬编码敏感信息 / No secrets hardcoded - [ ] 提交信息符合规范 / Commit message follows convention See merge request: cann/catccos!51	1 个月前
allgather_matmul_host.h	feat: 修改AGMM pytorch接入方式，从customClass改为TORCH_LIBRARY，并更新对应skill Co-authored-by: soyabean<sunyanbin8@huawei.com> # message auto-generated for no-merge-commit merge: !51 merge feat/agmm-torch-library-registration into master feat: 修改AGMM pytorch接入方式，从customClass改为TORCH_LIBRARY，并更新对应skill Created-by: i_soyabean Commit-by: soyabean Merged-by: cann-robot Description: ## 变更描述 / Description <!-- 本 PR 做了什么，为什么需要 / What does this PR do and why --> AGMM算子之前通过PyTorch custom class方式接入，算子无法入图。现采用TORCH_LIBRARY注册方式注册为PyTorch自定义算子，可以通过`torch.ops.catccos.*`调用算子。 - 新增AGMM Torch Library 接入路径。 - 删除了原来的通过custom class的接入方式。 - 更新了对应的`torch-binding` SKILL。 ## 改动类型 / Change Type - [ ] Bug 修复 / Bug Fix - [x] 新功能 / New Feature - [ ] 性能优化 / Performance - [ ] 代码重构 / Refactoring - [ ] 文档更新 / Documentation - [ ] 测试相关 / Test - [ ] 其它 / Other ## 关联 Issue / Related Issues <!-- Closes #000 可自动关闭 / Closes #000 to auto-close --> - Closes # - References # ## 测试信息 / Testing <!-- 简要测试说明或关键结果 / Brief test description or key results --> - [x] 单元测试通过 / UT passed - [x] 集成测试通过 / ST passed - [x] 人工验证通过 / Manual verified ## 检查清单 / Checklist - [ ] 代码符合规范 / Code follows style guide - [ ] 测试添加并通过 / Tests added and passed - [ ] 文档已更新 / Docs updated if needed - [ ] 无硬编码敏感信息 / No secrets hardcoded - [ ] 提交信息符合规范 / Commit message follows convention See merge request: cann/catccos!51	1 个月前
allgather_matmul_wrapper.cpp	feat: 修改AGMM pytorch接入方式，从customClass改为TORCH_LIBRARY，并更新对应skill Co-authored-by: soyabean<sunyanbin8@huawei.com> # message auto-generated for no-merge-commit merge: !51 merge feat/agmm-torch-library-registration into master feat: 修改AGMM pytorch接入方式，从customClass改为TORCH_LIBRARY，并更新对应skill Created-by: i_soyabean Commit-by: soyabean Merged-by: cann-robot Description: ## 变更描述 / Description <!-- 本 PR 做了什么，为什么需要 / What does this PR do and why --> AGMM算子之前通过PyTorch custom class方式接入，算子无法入图。现采用TORCH_LIBRARY注册方式注册为PyTorch自定义算子，可以通过`torch.ops.catccos.*`调用算子。 - 新增AGMM Torch Library 接入路径。 - 删除了原来的通过custom class的接入方式。 - 更新了对应的`torch-binding` SKILL。 ## 改动类型 / Change Type - [ ] Bug 修复 / Bug Fix - [x] 新功能 / New Feature - [ ] 性能优化 / Performance - [ ] 代码重构 / Refactoring - [ ] 文档更新 / Documentation - [ ] 测试相关 / Test - [ ] 其它 / Other ## 关联 Issue / Related Issues <!-- Closes #000 可自动关闭 / Closes #000 to auto-close --> - Closes # - References # ## 测试信息 / Testing <!-- 简要测试说明或关键结果 / Brief test description or key results --> - [x] 单元测试通过 / UT passed - [x] 集成测试通过 / ST passed - [x] 人工验证通过 / Manual verified ## 检查清单 / Checklist - [ ] 代码符合规范 / Code follows style guide - [ ] 测试添加并通过 / Tests added and passed - [ ] 文档已更新 / Docs updated if needed - [ ] 无硬编码敏感信息 / No secrets hardcoded - [ ] 提交信息符合规范 / Commit message follows convention See merge request: cann/catccos!51	1 个月前
main.cpp	feat: 修改AGMM pytorch接入方式，从customClass改为TORCH_LIBRARY，并更新对应skill Co-authored-by: soyabean<sunyanbin8@huawei.com> # message auto-generated for no-merge-commit merge: !51 merge feat/agmm-torch-library-registration into master feat: 修改AGMM pytorch接入方式，从customClass改为TORCH_LIBRARY，并更新对应skill Created-by: i_soyabean Commit-by: soyabean Merged-by: cann-robot Description: ## 变更描述 / Description <!-- 本 PR 做了什么，为什么需要 / What does this PR do and why --> AGMM算子之前通过PyTorch custom class方式接入，算子无法入图。现采用TORCH_LIBRARY注册方式注册为PyTorch自定义算子，可以通过`torch.ops.catccos.*`调用算子。 - 新增AGMM Torch Library 接入路径。 - 删除了原来的通过custom class的接入方式。 - 更新了对应的`torch-binding` SKILL。 ## 改动类型 / Change Type - [ ] Bug 修复 / Bug Fix - [x] 新功能 / New Feature - [ ] 性能优化 / Performance - [ ] 代码重构 / Refactoring - [ ] 文档更新 / Documentation - [ ] 测试相关 / Test - [ ] 其它 / Other ## 关联 Issue / Related Issues <!-- Closes #000 可自动关闭 / Closes #000 to auto-close --> - Closes # - References # ## 测试信息 / Testing <!-- 简要测试说明或关键结果 / Brief test description or key results --> - [x] 单元测试通过 / UT passed - [x] 集成测试通过 / ST passed - [x] 人工验证通过 / Manual verified ## 检查清单 / Checklist - [ ] 代码符合规范 / Code follows style guide - [ ] 测试添加并通过 / Tests added and passed - [ ] 文档已更新 / Docs updated if needed - [ ] 无硬编码敏感信息 / No secrets hardcoded - [ ] 提交信息符合规范 / Commit message follows convention See merge request: cann/catccos!51	1 个月前

使用方式

编译项目
进入示例目录并执行编译脚本：
```
cd examples/allgather_matmul
bash scripts/build.sh
```
运行AllGather-MatMul示例程序
在示例目录下执行运行脚本：
```
bash scripts/run.sh <device_list>
```
- 参数说明：
  - device_list：指定用于运行的设备（NPU）编号列表，以逗号分隔。
  - 示例：使用第6和第7个NPU设备运行2卡AllGather-MatMul示例：
```
bash scripts/run.sh 6,7
```
- 配置计算规模：
  矩阵形状参数（M、K、N）可在配置文件 scripts/test_shapes.csv 中进行设置。
  修改该文件以定义测试用例的输入维度。