Skip to content

Commit 96f3697

Browse files
Jerry-GeJerry-Ge
authored
[mlir][tosa] Update the description section for CastOp to align with TOSA v1.0 spec (#129958)
Updated the description section to include all data types and match the ordering with the spec. https://www.mlplatform.org/tosa/tosa_spec.html#_cast Signed-off-by: Jerry Ge <[email protected]>
1 parent ca582b1 commit 96f3697

File tree

1 file changed

+49
-19
lines changed

1 file changed

+49
-19
lines changed

‎mlir/include/mlir/Dialect/Tosa/IR/TosaOps.td‎

Lines changed: 49 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -2234,28 +2234,58 @@ def Tosa_CastOp: Tosa_Op<"cast", [Pure,
22342234
let summary = "Cast operation";
22352235

22362236
let description = [{
2237-
Performs a set of permissible cast operations
2237+
Casts a tensor from one data type to another.
2238+
* This table is showing the supported conversions from the TOSA Specification.
2239+
* The MLIR dialect here can be used to represent other conversions.
22382240

22392241
| Mode | Input | Output |
22402242
|--------------------------|---------|---------|
2241-
| signed 8 to bool | int8 | Boolean |
2242-
| signed 16 to bool | int16 | Boolean |
2243-
| signed 32 to bool | int32 | Boolean |
2244-
| bool to 8 | Boolean | int8 |
2245-
| bool to 16 | Boolean | int16 |
2246-
| bool to 32 | Boolean | int32 |
2247-
| signed 8 to signed 16 | int8 | int16 |
2248-
| signed 8 to signed 32 | int8 | int32 |
2249-
| signed 16 to signed 8 | int16 | int8 |
2250-
| signed 16 to signed 32 | int16 | int32 |
2251-
| signed 32 to signed 8 | int32 | int8 |
2252-
| signed 32 to signed 16 | int32 | int16 |
2253-
| float to signed 8 | float | int8 |
2254-
| float to signed 16 | float | int16 |
2255-
| signed 8 to float | int8 | float |
2256-
| signed 16 to float | int16 | float |
2257-
| float 32 to float 64 | float32 | float64 |
2258-
| float 64 to float 32 | float64 | float32 |
2243+
| fp16 to fp32 | float16 | float32 |
2244+
| fp16 to int 16 | float16 | int16 |
2245+
| fp16 to int 32 | float16 | int32 |
2246+
| fp16 to int 8 | float16 | int8 |
2247+
| fp32 to fp16 | float32 | float16 |
2248+
| fp32 to int 16 | float32 | int16 |
2249+
| fp32 to int 32 | float32 | int32 |
2250+
| fp32 to int 8 | float32 | int8 |
2251+
| int 16 to fp16 | int16 | float16 |
2252+
| int 16 to fp32 | int16 | float32 |
2253+
| int 32 to fp16 | int32 | float16 |
2254+
| int 32 to fp32 | int32 | float32 |
2255+
| int 8 to fp16 | int8 | float16 |
2256+
| int 8 to fp32 | int8 | float32 |
2257+
| bool to int 16 | Boolean | int16 |
2258+
| bool to int 32 | Boolean | int32 |
2259+
| bool to int 8 | Boolean | int8 |
2260+
| int 16 to bool | int16 | Boolean |
2261+
| int 16 to int 32 | int16 | int32 |
2262+
| int 16 to int 8 | int16 | int8 |
2263+
| int 32 to bool | int32 | Boolean |
2264+
| int 32 to int 16 | int32 | int16 |
2265+
| int 32 to int 8 | int32 | int8 |
2266+
| int 8 to bool | int8 | Boolean |
2267+
| int 8 to int 16 | int8 | int16 |
2268+
| int 8 to int 32 | int8 | int32 |
2269+
| bf16 to fp32 | bf16 | float32 |
2270+
| bf16 to int 16 | bf16 | int16 |
2271+
| bf16 to int 32 | bf16 | int32 |
2272+
| bf16 to int 8 | bf16 | int8 |
2273+
| fp32 to bf16 | float32 | bf16 |
2274+
| int 16 to bf16 | int16 | bf16 |
2275+
| int 32 to bf16 | int32 | bf16 |
2276+
| int 8 to bf16 | int8 | bf16 |
2277+
| bf16 to fp8e4m3 | bf16 | fp8e4m3 |
2278+
| fp8e4m3 to bf16 | fp8e4m3 | bf16 |
2279+
| bf16 to fp8e5m2 | bf16 | fp8e5m2 |
2280+
| fp8e5m2 to bf16 | fp8e5m2 | bf16 |
2281+
| fp16 to fp8e4m3 | float16 | fp8e4m3 |
2282+
| fp32 to fp8e4m3 | float32 | fp8e4m3 |
2283+
| fp8e4m3 to fp16 | fp8e4m3 | float16 |
2284+
| fp8e4m3 to fp32 | fp8e4m3 | float32 |
2285+
| fp16 to fp8e5m2 | float16 | fp8e5m2 |
2286+
| fp32 to fp8e5m2 | float32 | fp8e5m2 |
2287+
| fp8e5m2 to fp16 | fp8e5m2 | float16 |
2288+
| fp8e5m2 to fp32 | fp8e5m2 | float32 |
22592289
}];
22602290

22612291
let arguments = (ins

0 commit comments

Comments
 (0)