---
title: 函数说明
description: "KuDNN目前支持的算子如表1所示。"
url: https://www.hikunpeng.com/document/detail/zh/kunpenghpcs/hpckit/devg/KunpengHPCKit_developer_160.html
sourcePath: /source/zh/kunpenghpcs/hpckit/devg/KunpengHPCKit_developer_160.html
indexId: 705561b4fbba17aab3e26e6087643fca9f5c4d1f3a4f0e741427f2c208c3edae67
---
# 函数说明

KuDNN目前支持的算子如表1所示。


**表1 KuDNN支持的算子**

| 算子名称 | 算子说明 |
| --- | --- |
| GEMM（单线程）&Linear（多线程） | 矩阵乘法算子。 |
| Linear+postops | 在Linear算子计算后进行relu、silu、gelu、resext、resmul及residential等操作。 |
| Softmax | Softmax归一化算子。 |
| Conv2d/Conv3d | 2d/3d卷积算子。 |
| Group Normalization | 组归一化算子。 |
| Layer Normalization | 层归一化算子。 |
| Embedding | 映射。 |
| Root Mean Square Normalization | 均方根归一化算子。 |
| Rotary Position Embedding | 旋转位置编码。 |


常用对象如表2所示说明。


**表2 KuDNN::TensorInfo初始化需要传入参数列表**

| 参数名称 | 数据类型 | 描述 |
| --- | --- | --- |
| shape | struct KuDNN::Shape | {d1, d2, d3, m, n}最高5d需要和layout对应。 |
| type | struct KuDNN::Element::Type | 枚举类型支持以下4种： KuDNN::Element::TypeT::F16 KuDNN::Element::TypeT::F32 KuDNN::Element::TypeT::BF16 KuDNN::Element::TypeT::S8 |
| layout | enum class KuDNN::Layout | 枚举类型，具体支持的排布参考各算子TensorInfo对象初始化时支持的layout类型表格。 |