dataflow.get_running_device_id

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

函数功能

UDF执行时获取当前UDF的运行device_id，该信息来源于data_flow_deploy_info.json中的logic_device_list配置。

函数原型

get_running_device_id()

参数说明

无

返回值

返回当前UDF的运行device_id，数据类型为int。

调用示例

import dataflow as df

@df.pyflow
def func():
    print('running device_id = ', df.get_running_device_id())
    return a + b

约束说明

需配合pyflow装饰器进行使用。