Intel MKL-DNN 编程模型中的属性机制详解

姬牧格Ivy

于 2025-06-10 09:19:20 发布

阅读量336

点赞数 5

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/gitblog_00803/article/details/148552653

Intel MKL-DNN 编程模型中的属性机制详解

oneDNN 项目地址: https://gitcode.com/gh_mirrors/mk/mkl-dnn

概述

在深度学习计算库 Intel MKL-DNN 中，属性（Attributes）是控制算子行为的重要机制。本文将深入解析属性在 MKL-DNN 中的设计原理、使用方法和具体应用场景，帮助开发者更好地利用这一特性优化计算性能。

属性在算子创建流程中的位置

MKL-DNN 的算子创建分为两个关键步骤：

创建算子描述符（Primitive Descriptor）：基于计算引擎、操作参数和属性构建
创建算子（Primitive）：仅基于算子描述符创建实际可执行的算子

属性在这一流程中扮演着关键角色，它允许开发者在算子创建阶段灵活地指定各种额外参数，从而精细控制算子的执行行为。

属性的核心特性

不可变性设计

属性采用了一种巧妙的设计模式：一旦算子描述符创建完成，属性就被"冻结"（内部拷贝），后续对原始属性的修改不会影响已创建的算子描述符。这种设计既保证了灵活性，又确保了执行期的稳定性。

默认行为

如果开发者不显式设置属性，系统会使用默认属性配置。在C API中可以通过传递NULL，在C++ API中则可以直接省略属性参数。

属性使用示例

C API 使用方式

dnnl_primitive_attr_t attr;
dnnl_primitive_attr_create(&attr);
dnnl_primitive_attr_set_SOMETHING(attr, params); // 设置特定属性
dnnl_primitive_attr_set_SOMETHING_ELSE(attr, other_params);

// 创建算子描述符
dnnl_eltwise_backward_primitive_desc_create(&op_pd, engine, ..., hint_fwd_pd, attr);

// 属性对象可以立即销毁
dnnl_primitive_attr_destroy(attr);

C++ API 使用方式

dnnl::primitive_attr attr;
attr.set_SOMETHING(params);
attr.set_SOMETHING_ELSE(params);

// 创建算子描述符
primitive::primitive_desc pd(..., attr);

// 属性对象自动销毁