Skip to content

Commit 07d764a

Browse files
Merge pull request #17 from InfiniTensor/cat-patch
修改格式及status
2 parents 71d361e + afb6fe0 commit 07d764a

File tree

8 files changed

+127
-127
lines changed

8 files changed

+127
-127
lines changed

infiniop/ops/gemm/README.md

Lines changed: 22 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -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

infiniop/ops/random_sample/README.md

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ $$ P^{\prime} = \{ p_0^{\prime}, p_1^{\prime}, \dots, p_{n-1}^{\prime} \}, \quad
1616
其中
1717

1818
- $p_i$ 是第 $i$ 个元素的原始概率;
19-
- Top-k 选择概率最高的前 `k`
19+
- Top-k 选择概率最高的前 `k`
2020
- Top-p 选择累计概率和达到设定阈值 `p` 的最小子集;
2121

2222
## 接口
@@ -51,7 +51,7 @@ infiniStatus_t infiniopRandomSample(
5151
- `probs`:
5252
概率分布数据。张量限制见[创建算子描述](#创建算子描述)部分;
5353
- `random_val`:
54-
随机数种子,一般通过 Uniform 分布产生,范围是 $[0,1]$
54+
随机数种子,一般通过 Uniform 分布产生,范围是 $[0,1]$
5555
- `topp`:
5656
top-p 采样阈值,使得采样只从靠前的概率和为 `topp` 的范围内进行,范围是 $[0,1]$。当 `topp` 为0时,采样退化为 **Argmax** 算子。当 `topp` 大于等于1时,不设置 top-p 阈值;
5757
- `topk`:
@@ -67,8 +67,8 @@ infiniStatus_t infiniopRandomSample(
6767

6868
<div style="background-color: lightblue; padding: 1px;"> 返回值:</div>
6969

70-
-`random_val``topp``topk`、或 `temperature` 超出范围返回 [`INFINI_STATUS_BAD_PARAM`]
71-
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_INSUFFICIENT_WORKSPACE`], [`INFINI_STATUS_BAD_DEVICE`], [`INFINI_STATUS_EXECUTION_FAILED`].
70+
-`random_val``topp``topk`、或 `temperature` 超出范围返回 [`INFINI_STATUS_BAD_PARAM`]
71+
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_INSUFFICIENT_WORKSPACE`], [`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`], [`INFINI_STATUS_INTERNAL_ERROR`].
7272

7373
### 创建算子描述
7474

@@ -84,7 +84,7 @@ infiniStatus_t infiniopCreateRandomSampleDescriptor(
8484
<div style="background-color: lightblue; padding: 1px;"> 参数:</div>
8585

8686
- `handle`:
87-
`infiniopHandle_t` 类型的硬件控柄。详情请看:[`InfiniopHandle_t`]
87+
`infiniopHandle_t` 类型的硬件控柄。详情请看:[`InfiniopHandle_t`]
8888
- `desc_ptr`:
8989
`infiniopRandomSampleDescriptor_t` 指针,指向将被初始化的算子描述符地址;
9090
- `result` - ${ dOut | (,) | (,) }$:
@@ -94,13 +94,13 @@ infiniStatus_t infiniopCreateRandomSampleDescriptor(
9494

9595
参数限制:
9696

97-
- `dT`: `Float16``Float32`
98-
- `dOut`: `Uint64`
97+
- `dT`: `Float16``Float32`
98+
- `dOut`: `Uint64`
9999
- `N`: N > 0;
100100

101101
<div style="background-color: lightblue; padding: 1px;"> 返回值:</div>
102102

103-
- [`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`].
103+
- [`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`].
104104

105105
### 计算额外工作空间
106106

@@ -120,7 +120,7 @@ infiniStatus_t infiniopGetRandomSampleWorkspaceSize(
120120

121121
<div style="background-color: lightblue; padding: 1px;"> 返回值:</div>
122122

123-
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_BAD_DEVICE`].
123+
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`].
124124

125125
### 销毁算子描述符
126126

@@ -137,7 +137,7 @@ infiniStatus_t infiniopDestroyRandomSampleDescriptor(
137137

138138
<div style="background-color: lightblue; padding: 1px;"> 返回值: </div>
139139

140-
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_DEVICE`].
140+
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`].
141141

142142
## 已知问题
143143

@@ -152,8 +152,8 @@ infiniStatus_t infiniopDestroyRandomSampleDescriptor(
152152
[`INFINI_STATUS_SUCCESS`]: /common/status/README.md#INFINI_STATUS_SUCCESS
153153
[`INFINI_STATUS_BAD_PARAM`]: /common/status/README.md#INFINI_STATUS_BAD_PARAM
154154
[`INFINI_STATUS_INSUFFICIENT_WORKSPACE`]: /common/status/README.md#INFINI_STATUS_INSUFFICIENT_WORKSPACE
155-
[`INFINI_STATUS_BAD_DEVICE`]: /common/status/README.md#INFINI_STATUS_BAD_DEVICE
156-
[`INFINI_STATUS_EXECUTION_FAILED`]: /common/status/README.md#INFINI_STATUS_EXECUTION_FAILED
155+
[`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`]: /common/status/README.md#INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
156+
[`INFINI_STATUS_INTERNAL_ERROR`]: /common/status/README.md#INFINI_STATUS_INTERNAL_ERROR
157157
[`INFINI_STATUS_BAD_TENSOR_SHAPE`]: /common/status/README.md#INFINI_STATUS_BAD_TENSOR_SHAPE
158158
[`INFINI_STATUS_BAD_TENSOR_DTYPE`]: /common/status/README.md#INFINI_STATUS_BAD_TENSOR_DTYPE
159159
[`INFINI_STATUS_BAD_TENSOR_STRIDES`]: /common/status/README.md#INFINI_STATUS_BAD_TENSOR_STRIDES

infiniop/ops/rearrange/README.md

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77

88
例如,要转置 $2 \times 3$ 矩阵:
99

10-
> - 符号表示数据元素
11-
> - 下标表示数据在存储空间中的位置
10+
> - 符号表示数据元素
11+
> - 下标表示数据在存储空间中的位置
1212
1313
$$
1414
src:
@@ -80,17 +80,17 @@ infiniStatus_t infiniopRearrange(
8080
<div style="background-color: lightblue; padding: 1px;"> 参数: </div>
8181

8282
- `desc`:
83-
已使用 `infiniopCreateRearrangeDescriptor()` 初始化的算子描述符
83+
已使用 `infiniopCreateRearrangeDescriptor()` 初始化的算子描述符
8484
- `dst`:
85-
计算输出结果
85+
计算输出结果
8686
- `src`:
87-
输入张量
87+
输入张量
8888
- `stream`:
89-
计算流/队列
89+
计算流/队列
9090

9191
<div style="background-color: lightblue; padding: 1px;"> 返回值:</div>
9292

93-
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_BAD_DEVICE`], [`INFINI_STATUS_EXECUTION_FAILED`].
93+
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`], [`INFINI_STATUS_INTERNAL_ERROR`].
9494

9595
### 创建算子描述
9696

@@ -108,24 +108,24 @@ infiniStatus_t infiniopCreateRearrangeDescriptor(
108108
- `handle`:
109109
`infiniopHandle_t` 类型的硬件控柄。详情请看:[`InfiniopHandle_t`]
110110
- `desc_ptr`:
111-
`infiniopCreateRearrangeDescriptor` 指针,指向将被初始化的算子描述符地址
111+
`infiniopCreateRearrangeDescriptor` 指针,指向将被初始化的算子描述符地址
112112
- `dst_desc` - $\{ dT | shape | strides_{dst} \}$:
113-
算子输出 `dst` 的张量描述
113+
算子输出 `dst` 的张量描述
114114
- `src_desc` - $\{ dT | shape | strides_{src} \}$:
115-
算子计算参数 `src` 的张量描述
115+
算子计算参数 `src` 的张量描述
116116

117117
<div style="background-color: lightblue; padding: 1px;"> 参数限制:</div>
118118

119119
参数限制:
120120

121-
- $dT$: 任意类型
122-
- $shape$: 任意形状
123-
- $strides_{dst}$: 任意布局,但长度不为 1 的维度不能含有 0 步长(广播)
124-
- $strides_{src}$: 任意布局
121+
- $dT$: 任意类型
122+
- $shape$: 任意形状
123+
- $strides_{dst}$: 任意布局,但长度不为 1 的维度不能含有 0 步长(广播)
124+
- $strides_{src}$: 任意布局
125125

126126
<div style="background-color: lightblue; padding: 1px;"> 返回值:</div>
127127

128-
- [`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`].
128+
- [`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`].
129129

130130
### 销毁算子描述符
131131

@@ -138,25 +138,25 @@ infiniStatus_t infiniopDestroyRearrangeDescriptor(
138138
<div style="background-color: lightblue; padding: 1px;"> 参数: </div>
139139

140140
- `desc`:
141-
输入。待销毁的算子描述符
141+
输入。待销毁的算子描述符
142142

143143
<div style="background-color: lightblue; padding: 1px;"> 返回值: </div>
144144

145-
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_DEVICE`].
145+
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`].
146146

147147
## 已知问题
148148

149149
### 平台限制
150150

151-
- 除 CPU 外,其他平台不支持 3 阶以上的张量
151+
- 除 CPU 外,其他平台不支持 3 阶以上的张量
152152

153153
<!-- 链接 -->
154154
[`InfiniopHandle_t`]: /infiniop/handle/README.md
155155

156156
[`INFINI_STATUS_SUCCESS`]:/common/status/README.md#INFINI_STATUS_SUCCESS
157157
[`INFINI_STATUS_BAD_PARAM`]:/common/status/README.md#INFINI_STATUS_BAD_PARAM
158-
[`INFINI_STATUS_BAD_DEVICE`]:/common/status/README.md#INFINI_STATUS_BAD_DEVICE
159-
[`INFINI_STATUS_EXECUTION_FAILED`]:/common/status/README.md#INFINI_STATUS_EXECUTION_FAILED
158+
[`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`]:/common/status/README.md#INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
159+
[`INFINI_STATUS_INTERNAL_ERROR`]:/common/status/README.md#INFINI_STATUS_INTERNAL_ERROR
160160
[`INFINI_STATUS_BAD_TENSOR_SHAPE`]:/common/status/README.md#INFINI_STATUS_BAD_TENSOR_SHAPE
161161
[`INFINI_STATUS_BAD_TENSOR_DTYPE`]:/common/status/README.md#INFINI_STATUS_BAD_TENSOR_DTYPE
162162
[`INFINI_STATUS_BAD_TENSOR_STRIDES`]:/common/status/README.md#INFINI_STATUS_BAD_TENSOR_STRIDES

infiniop/ops/rms_norm/README.md

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -34,25 +34,25 @@ infiniStatus_t infiniopRMSNorm(
3434
<div style="background-color: lightblue; padding: 1px;"> 参数: </div>
3535

3636
- `desc`:
37-
使用 `infiniopCreateRMSNormDescriptor()` 初始化的算子描述符
37+
使用 `infiniopCreateRMSNormDescriptor()` 初始化的算子描述符
3838
- `workspace`:
39-
算子计算所需的额外工作空间
39+
算子计算所需的额外工作空间
4040
- `workspace_size`:
41-
`workspace` 的大小,单位:字节
41+
`workspace` 的大小,单位:字节
4242
- `y`:
43-
计算结果地址。张量限制见[创建算子描述](#创建算子描述)部分
43+
计算结果地址。张量限制见[创建算子描述](#创建算子描述)部分
4444
- `x`:
45-
输入数据地址。张量限制见[创建算子描述](#创建算子描述)部分
45+
输入数据地址。张量限制见[创建算子描述](#创建算子描述)部分
4646
- `w`:
47-
权重数据地址(可选)。如果权重张量为 `NULL`,则不进行缩放
47+
权重数据地址(可选)。如果权重张量为 `NULL`,则不进行缩放
4848
- `stream`:
49-
计算流/队列
49+
计算流/队列
5050

5151
<div style="background-color: lightblue; padding: 1px;"> 返回值:</div>
5252

5353
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_INSUFFICIENT_WORKSPACE`], [`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`], [`INFINI_STATUS_INTERNAL_ERROR`].
5454

55-
-`epsilon` 超出范围时返回 [`INFINI_STATUS_BAD_PARAM`]
55+
-`epsilon` 超出范围时返回 [`INFINI_STATUS_BAD_PARAM`]
5656

5757
### 创建算子描述
5858

@@ -70,22 +70,22 @@ infiniStatus_t infiniopCreateRMSNormDescriptor(
7070
<div style="background-color: lightblue; padding: 1px;"> 参数:</div>
7171

7272
- `handle`:
73-
硬件控柄。详情请看:[`InfiniopHandle_t`]
73+
硬件控柄。详情请看:[`InfiniopHandle_t`]
7474
- `desc_ptr`:
75-
存放将被初始化的算子描述符地址
75+
存放将被初始化的算子描述符地址
7676
- `y_desc` - { dT | (b, d) | (..., 1) }:
77-
计算结果 `y` 的张量描述
77+
计算结果 `y` 的张量描述
7878
- `x_desc` - { dT | (b, d) | (..., 1) }:
79-
输入 `x` 的张量描述,形状和 `y_desc` 保持一致
79+
输入 `x` 的张量描述,形状和 `y_desc` 保持一致
8080
- `w_desc` - { dW | (d,) | (1,) }:
81-
权重 `w` 的张量描述。`w_desc` 为一维张量,长度和归一化维度的长度保持一致。如果权重张量为 `NULL`,则不进行缩放
81+
权重 `w` 的张量描述。`w_desc` 为一维张量,长度和归一化维度的长度保持一致。如果权重张量为 `NULL`,则不进行缩放
8282
- `epsilon` - float:
83-
用于避免除以零的小常数,范围是 $(0, 1]$
83+
用于避免除以零的小常数,范围是 $(0, 1]$
8484

8585
参数限制:
8686

87-
- `dT`: (`Float16`, `Float32`) 之一
88-
- `dW`: 当 `dT``Float16`时,为(`Float16`, `Float32`) 之一`dT``Float32`时,为 `Float32`
87+
- `dT`: (`Float16`, `Float32`) 之一
88+
- `dW`: 当 `dT``Float16`时,为(`Float16`, `Float32`) 之一`dT``Float32`时,为 `Float32`
8989

9090
<div style="background-color: lightblue; padding: 1px;"> 返回值:</div>
9191

@@ -103,13 +103,13 @@ infiniStatus_t infiniopGetRMSNormWorkspaceSize(
103103
<div style="background-color: lightblue; padding: 1px;"> 参数:</div>
104104

105105
- `desc`:
106-
使用 `infiniopCreateRMSNormDescriptor()` 初始化的算子描述符
106+
使用 `infiniopCreateRMSNormDescriptor()` 初始化的算子描述符
107107
- `size`:
108-
存放空间大小的计算结果的地址
108+
存放空间大小的计算结果的地址
109109

110110
<div style="background-color: lightblue; padding: 1px;"> 返回值:</div>
111111

112-
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_BAD_DEVICE`].
112+
- [`INFINI_STATUS_SUCCESS`], [`INFINI_STATUS_BAD_PARAM`], [`INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED`].
113113

114114
### 销毁算子描述符
115115

@@ -122,7 +122,7 @@ infiniStatus_t infiniopDestroyRMSNormDescriptor(
122122
<div style="background-color: lightblue; padding: 1px;"> 参数: </div>
123123

124124
- `desc`:
125-
待销毁的算子描述符
125+
待销毁的算子描述符
126126

127127
<div style="background-color: lightblue; padding: 1px;"> 返回值: </div>
128128

0 commit comments

Comments
 (0)