SetCompileConfig（GraphPp类）
【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
函数功能
设置GraphPp的json配置文件路径和文件名。配置文件用于AscendGraph的描述和编译。
返回设置好的GraphPp。
函数原型
GraphPp &SetCompileConfig(const char_t *json_file_path)参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| json_file_path | 输入 | GraphPp的json配置文件路径和文件名。 GraphPp的json配置文件用于AscendGraph的描述和编译。 示例如下，参数解释请参考表1。 {"build_option":{},"inputs_tensor_desc":[{"data_type":"DT_UINT32","shape":[3]},{"data_type":"DT_UINT32","shape":[3]}]} |
表 1GraphPp的json配置文件
| 配置项 | 可选/必选 | 描述 |
|---|---|---|
| build_option | 可选 | 值为map<string, string>, 有需要设置时参考Ascend Graph中的build_option。 |
| inputs_tensor_desc | 可选 | 值为list，Ascend Graph的输入节点，list元素为Tensor的描述。 |
| inputs_tensor_desc.data_type | 可选 | 字符串类型。 取值为Ascend Graph中的data_type对应字符串。 |
| inputs_tensor_desc.shape | 可选 | 值为整数类型的列表 取值为Ascend Graph中的shape。 |
返回值
返回设置好的GraphPp。
异常处理
无。
约束说明
无。
【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
