GeApi
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
模块导入
from ge.ge_global import GeApi
功能说明
GeApi 类提供 GE 的全局初始化和资源释放接口。ge_initialize 必须在使用其他 GE 接口之前调用,ge_finalize 在程序退出时调用以释放资源。两个方法都是类方法,无需实例化 GeApi 即可调用。
类定义
class GeApi:
@classmethod
def ge_initialize(cls, config: dict) -> None
@classmethod
def ge_finalize(cls) -> None
函数说明
ge_initialize
@classmethod
def ge_initialize(cls, config: dict) -> None
功能说明:初始化 GE,准备执行环境。此方法为类方法,无需实例化 GeApi 即可调用。必须在使用其他 GE 接口之前调用。
参数说明:
| 参数名 | 类型 | 必选/可选 | 说明 |
|---|---|---|---|
| config | dict | 必选 | GE 初始化配置字典,键值对均为字符串类型,用于指定 GE 的运行参数。 |
返回值说明:无返回值。
约束说明:
- config 必须为 dict 类型,否则抛出 TypeError。
- config 不能为空字典,否则抛出 TypeError。
- GE 初始化失败时将抛出 RuntimeError。
- 必须在使用其他 GE 接口之前调用此方法。
- 请勿重复调用 ge_initialize,重复调用可能导致未定义行为。
ge_finalize
@classmethod
def ge_finalize(cls) -> None
功能说明:释放 GE 资源,清理执行环境。此方法为类方法,无需实例化 GeApi 即可调用。应在程序退出或不再需要 GE 时调用。
参数说明:无参数。
返回值说明:无返回值。
约束说明:
- 在调用 ge_finalize 之前,须确保所有 Session 已销毁,所有图资源已释放。
- GE 资源释放失败时将抛出 RuntimeError。
- 请勿在 ge_finalize 之后继续使用任何 GE 接口。