Merge pull request #34 from InfiniTensor/update_bf16

更新算子列表

Author: PanZezhong1725

infiniop/ops/README.md

@@ -2,10 +2,13 @@
 
 - [`Add`](/infiniop/ops/add/README.md)
 - [`Causal Softmax`](/infiniop/ops/causal_softmax/README.md)
+- [`Clip`](/infiniop/ops/clip/README.md)
 - [`GEMM`](/infiniop/ops/gemm/README.md)
+- [`Mul`](/infiniop/ops/mul/README.md)
 - [`Random Sample`](/infiniop/ops/random_sample/README.md)
 - [`Rearrange`](/infiniop/ops/rearrange/README.md)
 - [`RMS Norm`](/infiniop/ops/rms_norm/README.md)
 - [`RoPE`](/infiniop/ops/rope/README.md)
 - [`Softmax`](/infiniop/ops/softmax/README.md)
+- [`Sub`](/infiniop/ops/sub/README.md)
 - [`SwiGLU`](/infiniop/ops/swiglu/README.md)

infiniop/ops/add/README.md

@@ -71,7 +71,7 @@ infiniStatus_t infiniopCreateAddDescriptor(
 
 参数限制：
 
-- `dT`:  (`Float16`, `Float32`, `Float64`) 之一。
+- `dT`:  (`Float16`, `Float32`, `Float64`, `BFloat16`) 之一。
 - 输入 `a` 与 `b` 的形状需与 `c` 相同。`a` 与 `b` 涉及多向广播时需调整步长以匹配多向广播的映射关系。
 - 支持原位计算，即计算时 `c` 可以和 `a` 或 `b` 指向同一地址。
 - 计算输出参数 `c` 不能进行广播（`c` 的步长不能涉及广播设置，即步长不能有 0）

infiniop/ops/causal_softmax/README.md

@@ -76,7 +76,7 @@ infiniStatus_t infiniopCreateCausalSoftmaxDescriptor(
 
 参数限制：
 
- - **`dT`**:  (`Float16`, `Float32`, `Double`, `Bfloat16`) 之一
+ - **`dT`**:  (`Float16`, `Float32`, `Float64`, `Bfloat16`) 之一
 
 <div style="background-color: lightblue; padding: 1px;"> 返回值：</div>
 

infiniop/ops/clip/README.md

@@ -85,8 +85,12 @@ infiniStatus_t infiniopCreateClipDescriptor(
 - `max_val` - {dT | (d1, ..., dn) | (...)} :
   裁剪的最大值张量描述，必须是与输入张量形状相同的张量。
 
+参数限制：
 
-
+- `dT`:  (`Float16`, `Float32`, `Float64`, `BFloat16`) 之一。
+- 输入 `a` 与 `b` 的形状需与 `c` 相同。`a` 与 `b` 涉及多向广播时需调整步长以匹配多向广播的映射关系。
+- 支持原位计算，即计算时 `c` 可以和 `a` 或 `b` 指向同一地址。
+- 计算输出参数 `c` 不能进行广播（`c` 的步长不能涉及广播设置，即步长不能有 0）
 
 <div style="background-color: lightblue; padding: 1px;"> 返回值：</div>
 

infiniop/ops/gemm/README.md

@@ -84,7 +84,7 @@ infiniStatus_t infiniopCreateGemmDescriptor(
 
 参数限制：
 
-- `dT`: (`Float16`, `Float32`) 之一；
+- `dT`: (`Float16`, `Float32`, `BFloat16`) 之一；
 - `[batch,]`: `batch ≥ 1`（可选）；
 - `M`: M > 0；
 - `N`: N > 0；

infiniop/ops/mul/README.md

@@ -70,7 +70,7 @@ infiniStatus_t infiniopCreateMulDescriptor(
 
 参数限制：
 
-- `dT`:  (`Float16`, `Float32`, `Float64`) 之一。
+- `dT`:  (`Float16`, `Float32`, `Float64`, `BFloat16`) 之一。
 - 输入 `a` 与 `b` 的形状需与 `c` 相同。`a` 与 `b` 涉及多向广播时需调整步长以匹配多向广播的映射关系。
 - 支持原位计算，即计算时 `c` 可以和 `a` 或 `b` 指向同一地址。
 - 计算输出参数 `c` 不能进行广播（`c` 的步长不能涉及广播设置，即步长不能有 0）

infiniop/ops/random_sample/README.md

@@ -94,7 +94,7 @@ infiniStatus_t infiniopCreateRandomSampleDescriptor(
 
 参数限制：
 
-- `dT`: `Float16` 或 `Float32`。
+- `dT`: (`Float16`, `Float32`, `BFloat16`) 之一。
 - `dOut`: `Uint64`。
 - `N`: N > 0；
 

infiniop/ops/rms_norm/README.md

@@ -84,8 +84,8 @@ infiniStatus_t infiniopCreateRMSNormDescriptor(
 
 参数限制：
 
-- `dT`: (`Float16`, `Float32`) 之一。
-- `dW`: 当 `dT` 为 `Float16`时，为(`Float16`, `Float32`) 之一。当 `dT` 为 `Float32`时，为 `Float32`。
+- `dT`: (`Float16`, `Float32`, `BFloat16`) 之一。
+- `dW`: 与 `dT` 相同，或为 `Float32`。
 
 <div style="background-color: lightblue; padding: 1px;"> 返回值：</div>
 

infiniop/ops/softmax/README.md

@@ -55,7 +55,7 @@ infiniStatus_t infiniopCreateSoftmaxDescriptor(
 
 参数限制：
 
-- **`dT`**:  (`Float16`, `Float32`, `Double`) 之一。
+- **`dT`**:  (`Float16`, `Float32`, `Float64`) 之一。
 
 <div style="background-color: lightblue; padding: 1px;"> 返回值：</div>
 

infiniop/ops/sub/README.md

@@ -71,7 +71,7 @@ infiniStatus_t infiniopCreateSubDescriptor(
 
 参数限制：
 
-- `dT`:  (`Float16`, `Float32`, `Float64`) 之一。
+- `dT`:  (`Float16`, `Float32`, `Float64`, `BFloat16`) 之一。
 - 输入 `a` 与 `b` 的形状需与 `c` 相同。`a` 与 `b` 涉及多向广播时需调整步长以匹配多向广播的映射关系。
 - 支持原位计算，即计算时 `c` 可以和 `a` 或 `b` 指向同一地址。
 - 计算输出参数 `c` 不能进行广播（`c` 的步长不能涉及广播设置，即步长不能有 0）