通信类API数据可以使用全量dump方式获取,若只dump通信类API数据,可以使用如下示例:
debugger.configure_hook(mode="api_list", api_list=["distributed"])
通信类API支持列表:
序号 |
Distributed |
---|---|
1 |
send |
2 |
recv |
3 |
broadcast |
4 |
all_reduce |
5 |
reduce |
6 |
all_gather |
7 |
gather |
8 |
isend |
9 |
irecv |
10 |
scatter |
11 |
reduce_scatter |
12 |
_reduce_scatter_base |
13 |
_all_gather_base |