已合并
PTO-COMM-ISA #17
zhezhou创建于 2025年12月30日
PTO-COMM-ISA #17
已合并
zhezhou创建于 2025年12月30日
zhezhou
zhezhou
2025年12月30日

描述

定义并且实现PTO-COMM-ISA,Tile粒度的多卡P2P/同步/P2MP操作 定义统一的抽象,并且提供实现。

Point-to-Point Communication (Synchronous)

  • TPUT: docs/isa/comm/TPUT.md - Remote write (GM → UB → GM)
  • TGET: docs/isa/comm/TGET.md - Remote read (GM → UB → GM)

Signal-Based Synchronization

  • TNOTIFY: docs/isa/comm/TNOTIFY.md - Send notification to remote NPU
  • TWAIT: docs/isa/comm/TWAIT.md - Blocking wait for signal condition
  • TTEST: docs/isa/comm/TTEST.md - Non-blocking test signal condition

Collective Communication

  • TGATHER: docs/isa/comm/TGATHER.md - Gather data from all ranks
  • TSCATTER: docs/isa/comm/TSCATTER.md - Scatter data to all ranks
  • TREDUCE: docs/isa/comm/TREDUCE.md - Reduce data from all ranks to local
  • TBROADCAST: docs/isa/comm/TBROADCAST.md - Broadcast from current NPU to all ranks

测试

  1. 安装cann-shmem 并且set执行环境 https://gitcode.com/cann/shmem?source_module=search_result_repo
  2. bash tests/run_comm_test.sh 可以运行全部 comm-isa 测试用例

文档更新

isa/comm/README.md 文件

类型标签

likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 zhezhou 的贡献)
CANN-robot
CANN-robot成员
2025年12月30日 评论:

Thank your for your pull-request.

The full list of commands accepted by me can be found at here.

You can get sig-info at here

likedislike
CANN-robot
CANN-robot成员
2025年12月30日 评论:

以下是根据您提交的修改文件推荐的Reviewer和Committer序列,需各模块评审通过后方可合入

Module List Reviewers Committers
repo-cann/pto-isa csjlchen, Zhanghaijian, qukelin, bluesky901, changxianyu changxianyu, ChanKaLok, HuangXingYuan_777, csjlchen, Zhanghaijian
likedislike
CANN-robotCANN-robot成员
2025年12月30日 将csjlchen,bluesky901,changxianyu,Zhanghaijian,HuangXingYuan_777,qukelin,ChanKaLok设为审查人
CANN-robotCANN-robot成员
2025年12月30日 将csjlchen,bluesky901,changxianyu,Zhanghaijian,HuangXingYuan_777,qukelin,ChanKaLok设为评审人
CANN-robotCANN-robot成员
2025年12月30日 添加了label:cann-cla/no
此处折叠了214条消息 查看更多
CANN-robot
CANN-robot成员
2月12日 评论:

The following label is not ready.

lgtm: Please wait for reviewers to review the code.

likedislike
Zhanghaijian成员
2月12日 评论:

/lgtm

likedislike
CANN-robotCANN-robot成员
2月12日 添加了label:lgtm
CANN-robot
CANN-robot成员
2月12日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: csjlchen, HuangXingYuan_777, SherwinBai, zhezhou.
Reviewers who wrote a comment of /lgtm are: Zhanghaijian, zhezhou, csjlchen, HuangXingYuan_777, SherwinBai.

likedislike
CANN-robotCANN-robot成员
2月12日 合入了pull request