@@ -7,9 +7,9 @@ $$ C = α ⋅ (A * B) + β ⋅ C $$
77
88其中:
99
10- - ` A ` 为左输入张量,形状为 ` ( [batch,] M, K ) ` ;
11- - ` B ` 为右输入张量,形状为 ` ( [batch,] K, N ) ` ;
12- - ` C ` 的形状由矩阵乘法规则确定,形状为 ` ( [batch,] M, N ) ` ;
10+ - ` A ` 为左输入张量,形状为 ` ( [batch,] M, K ) ` 。
11+ - ` B ` 为右输入张量,形状为 ` ( [batch,] K, N ) ` 。
12+ - ` C ` 的形状由矩阵乘法规则确定,形状为 ` ( [batch,] M, N ) ` 。
1313- ` α ` 为缩放因子,` β ` 为累加系数;
1414
1515## 接口
@@ -31,23 +31,23 @@ infiniStatus_t infiniopGemm(
3131<div style =" background-color : lightblue ; padding : 1px ;" > 参数: </div >
3232
3333- ` desc ` :
34- 已使用 ` infiniopCreateGemmDescriptor() ` 初始化的算子描述符;
34+ 已使用 ` infiniopCreateGemmDescriptor() ` 初始化的算子描述符。
3535- ` workspace ` :
36- 指向算子计算所需的额外工作空间;
36+ 指向算子计算所需的额外工作空间。
3737- ` workspace_size ` :
38- ` workspace ` 的大小,单位:字节;
38+ ` workspace ` 的大小,单位:字节。
3939- ` c ` :
40- 计算输出结果。张量限制见[ 创建算子描述] ( #创建算子描述 ) 部分;
40+ 计算输出结果。张量限制见[ 创建算子描述] ( #创建算子描述 ) 部分。
4141- ` a ` :
42- 左输入张量。张量限制见[ 创建算子描述] ( #创建算子描述 ) 部分;
42+ 左输入张量。张量限制见[ 创建算子描述] ( #创建算子描述 ) 部分。
4343- ` b ` :
44- 右输入张量。张量限制见[ 创建算子描述] ( #创建算子描述 ) 部分;
44+ 右输入张量。张量限制见[ 创建算子描述] ( #创建算子描述 ) 部分。
4545- ` stream ` :
46- 计算流/队列;
46+ 计算流/队列。
4747
4848<div style =" background-color : lightblue ; padding : 1px ;" > 返回值:</div >
4949
50- - [ ` INFINI_STATUS_SUCCESS ` ] , [ ` INFINI_STATUS_BAD_PARAM ` ] , [ ` INFINI_STATUS_INSUFFICIENT_WORKSPACE ` ] , [ ` INFINI_STATUS_BAD_DEVICE ` ] , [ ` INFINI_STATUS_EXECUTION_FAILED ` ] .
50+ - [ ` INFINI_STATUS_SUCCESS ` ] , [ ` INFINI_STATUS_BAD_PARAM ` ] , [ ` INFINI_STATUS_INSUFFICIENT_WORKSPACE ` ] , [ ` INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED ` ] , [ ` INFINI_STATUS_INTERNAL_ERROR ` ] .
5151
5252### 创建算子描述
5353
@@ -70,15 +70,15 @@ infiniStatus_t infiniopCreateGemmDescriptor(
7070- ` desc_ptr ` :
7171 指向将被初始化的算子描述符地址;
7272- ` c_desc ` - { dT | ( [ batch,] , M, N) | (~ ) }:
73- 算子计算参数 ` c ` 的张量描述;
73+ 算子计算参数 ` c ` 的张量描述。
7474- ` a_desc ` - { dT | ( [ batch,] , M, K) | (~ ) }:
75- 算子计算参数 ` a ` 的张量描述;
75+ 算子计算参数 ` a ` 的张量描述。
7676- ` b_desc ` - { dT | ( [ batch,] , K, N) | (~ ) }:
77- 算子计算参数 ` b ` 的张量描述;
77+ 算子计算参数 ` b ` 的张量描述。
7878- ` alpha ` - float:
79- 算子计算缩放因子;
79+ 算子计算缩放因子。
8080- ` beta ` - float:
81- 算子计算累加系数;
81+ 算子计算累加系数。
8282
8383<div style =" background-color : lightblue ; padding : 1px ;" > 参数限制:</div >
8484
@@ -92,7 +92,7 @@ infiniStatus_t infiniopCreateGemmDescriptor(
9292
9393<div style =" background-color : lightblue ; padding : 1px ;" > 返回值:</div >
9494
95- - [ ` INFINI_STATUS_SUCCESS ` ] , [ ` INFINI_STATUS_BAD_PARAM ` ] , [ ` INFINI_STATUS_BAD_TENSOR_SHAPE ` ] , [ ` INFINI_STATUS_BAD_TENSOR_DTYPE ` ] , [ ` INFINI_STATUS_BAD_TENSOR_STRIDES ` ] , [ ` INFINI_STATUS_BAD_DEVICE ` ] .
95+ - [ ` INFINI_STATUS_SUCCESS ` ] , [ ` INFINI_STATUS_BAD_PARAM ` ] , [ ` INFINI_STATUS_BAD_TENSOR_SHAPE ` ] , [ ` INFINI_STATUS_BAD_TENSOR_DTYPE ` ] , [ ` INFINI_STATUS_BAD_TENSOR_STRIDES ` ] , [ ` INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED ` ] .
9696
9797### 计算额外工作空间
9898
@@ -106,13 +106,13 @@ infiniStatus_t infiniopGetGemmWorkspaceSize(
106106<div style =" background-color : lightblue ; padding : 1px ;" > 参数:</div >
107107
108108- ` desc ` :
109- 已使用 ` infiniopCreateMatmulDescriptor() ` 初始化的算子描述符;
109+ 已使用 ` infiniopCreateMatmulDescriptor() ` 初始化的算子描述符。
110110- ` size ` :
111111 额外空间大小的计算结果的写入地址;
112112
113113<div style =" background-color : lightblue ; padding : 1px ;" > 返回值:</div >
114114
115- - [ ` INFINI_STATUS_SUCCESS ` ] , [ ` INFINI_STATUS_BAD_PARAM ` ] , [ ` INFINI_STATUS_BAD_DEVICE ` ] .
115+ - [ ` INFINI_STATUS_SUCCESS ` ] , [ ` INFINI_STATUS_BAD_PARAM ` ] , [ ` INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED ` ] .
116116
117117### 销毁算子描述符
118118
@@ -129,16 +129,16 @@ infiniStatus_t infiniopDestroyGemmDescriptor(
129129
130130<div style =" background-color : lightblue ; padding : 1px ;" > 返回值: </div >
131131
132- - [ ` INFINI_STATUS_SUCCESS ` ] , [ ` INFINI_STATUS_BAD_DEVICE ` ] .
132+ - [ ` INFINI_STATUS_SUCCESS ` ] , [ ` INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED ` ] .
133133
134134<!-- 链接 -->
135135[ `InfiniopHandle_t` ] : /infiniop/handle/README.md
136136
137137[ `INFINI_STATUS_SUCCESS` ] :/common/status/README.md#INFINI_STATUS_SUCCESS
138138[ `INFINI_STATUS_BAD_PARAM` ] :/common/status/README.md#INFINI_STATUS_BAD_PARAM
139139[ `INFINI_STATUS_INSUFFICIENT_WORKSPACE` ] :/common/status/README.md#INFINI_STATUS_INSUFFICIENT_WORKSPACE
140- [ `INFINI_STATUS_BAD_DEVICE ` ] :/common/status/README.md#INFINI_STATUS_BAD_DEVICE
141- [ `INFINI_STATUS_EXECUTION_FAILED ` ] :/common/status/README.md#INFINI_STATUS_EXECUTION_FAILED
140+ [ `INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED ` ] :/common/status/README.md#INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
141+ [ `INFINI_STATUS_INTERNAL_ERROR ` ] :/common/status/README.md#INFINI_STATUS_INTERNAL_ERROR
142142[ `INFINI_STATUS_BAD_TENSOR_SHAPE` ] :/common/status/README.md#INFINI_STATUS_BAD_TENSOR_SHAPE
143143[ `INFINI_STATUS_BAD_TENSOR_DTYPE` ] :/common/status/README.md#INFINI_STATUS_BAD_TENSOR_DTYPE
144144[ `INFINI_STATUS_BAD_TENSOR_STRIDES` ] :/common/status/README.md#INFINI_STATUS_BAD_TENSOR_STRIDES
0 commit comments