---
title: CustomAllocateStreamPassFn
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/ascendgraphapi/atlasgeapi_07_0154.html
sourcePath: /source/zh/canncommercial/900/API/ascendgraphapi/atlasgeapi_07_0154.html
indexId: 8cb4acb09c648c4b1c2c301f39f60134d3cc1f923412434fe266ff3a20f9142472
---
# CustomAllocateStreamPassFn

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | √ |
| Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |


#### 头文件/库文件

- 头文件：#include <register/register_custom_pass.h>
- 库文件：libregister.so


#### 功能说明

注册自定义的逻辑流分配Pass执行函数。

关于接口的详细使用方法请参见使用自定义逻辑流分配Pass定制并发(https://www.hiascend.comdocument/detail/zh/canncommercial/900/programug/graphdevg/atlasag_25_0076.html)。


#### 函数原型

```
PassRegistrationData &CustomAllocateStreamPassFn(const CustomAllocateStreamPassFunc &allocate_stream_pass_fn)
```


#### 参数说明

| 参数名 | 输入/输出 | 说明 |
| --- | --- | --- |
| allocate\_stream\_pass\_fn | 输入 | 自定义的逻辑流分配函数。详情请参见回调函数CustomAllocateStreamPassFunc。 |


#### 返回值说明

返回自身对象的引用。


#### 约束说明

无。


#### 回调函数CustomAllocateStreamPassFunc

用户自定义并实现CustomAllocateStreamPassFunc类函数，即自定义的逻辑流分配函数。

```
Status CustomAllocateStreamPassFunc(const ConstGraphPtr &graph, StreamPassContext &stream_context)
```


**表1 参数说明**

| 参数名 | 输入/输出 | 说明 |
| --- | --- | --- |
| graph | 输入 | 待分配逻辑流的图 |
| stream\_context | 输入 | 逻辑流分配上下文，可通过该上下文申请新stream id，设置节点的stream id等。详见StreamPassContext结构定义。 |
| \- | 输出 | SUCCESS：成功。 其他值：失败。 |