CustomAllocateStreamPassFn
Applicability
|
Product |
Supported |
|---|---|
|
Atlas 350 Accelerator Card |
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
Header File/Library File
- Header file: #include <register/register_custom_pass.h>
- Library file: libregister.so
Function Usage
Registers the execution function for a custom logical stream allocation pass.
For details about the API, see Custom Concurrency Using CustomStreamPassFunc.
Prototype
1
|
PassRegistrationData &CustomAllocateStreamPassFn(const CustomAllocateStreamPassFunc &allocate_stream_pass_fn) |
Parameters
|
Parameter |
Input/Output |
Description |
|---|---|---|
|
allocate_stream_pass_fn |
Input |
Custom logical stream allocation function. For details, see Callback Function CustomAllocateStreamPassFunc. |
Returns
Reference to the object itself.
Restrictions
None
Callback Function CustomAllocateStreamPassFunc
You can customize and implement the CustomAllocateStreamPassFunc class function.
Status CustomAllocateStreamPassFunc(const ConstGraphPtr &graph, StreamPassContext &stream_context)
|
Parameter |
Input/Output |
Description |
|---|---|---|
|
graph |
Input |
Graph for which logical streams are to be allocated. |
|
stream_context |
Input |
Logical stream allocation context, which can be used to request new stream IDs, set stream IDs for nodes, and perform other stream control operations. For details, see the structure definition of StreamPassContext. |
|
- |
Output |
|