昇腾/GE DFlow API set_attr函数
set_attr
【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
函数功能
设置FlowNode的属性。
函数原型
set_attr(attr_name, attr_value)参数说明
| 参数名称 | 数据类型 | 取值说明 |
|---|---|---|
| attr_name | str | 属性名称 当前支持如下三种： - _flow_attr：BOOL类型属性，可以配置为true或false，true表示存在flow属性，false表示不存在flow属性。 -_flow_attr_depth：INT类型属性，指定了队列的深度，范围：大于等于2，若不配置，默认深度为128。 - _flow_attr_enqueue_policy：STRING类型属性，指定的入队的策略，仅支持"FIFO"和"OVERWRITE"，"FIFO"表示队列数据顺序入队，队列满的时候会等待dequeue，"OVERWRITE"表示入队不会等待，数据会循环覆盖，若不配置，默认策略为FIFO。 |
| attr_value | Union[bool, str, int] | 属性值。 |
返回值
正常场景下返回None。
返回“TypeError”表示参数类型不正确。
调用示例
import dataflow as df pp = df.FuncProcessPoint(...) flow_node = df.FlowNode(...) flow_node.set_attr("_flow_attr",True) flow_node.set_attr("_flow_attr_depth",5) flow_node.set_attr("_flow_attr_enqueue_policy","FIFO")约束说明
通过_flow_attr_depth和_flow_attr_enqueue_policy设置depth和policy需要先使能_flow_attr，否则不会生效。
【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考