[Mlir-commits] [mlir] [mlir][tosa] Change CONV2D type combinations for BF16 & FP16 output (PR #215611)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Tue Aug 11 10:02:23 PDT 2026
llvmorg-github-actions[bot] wrote:
<!--LLVM PR SUMMARY COMMENT-->
@llvm/pr-subscribers-mlir
Author: Jeremy Johnson (jjohnson-arm)
<details>
<summary>Changes</summary>
Note that this implements the TOSA specification change: https://github.com/arm/tosa-specification/pull/77
---
Patch is 84.98 KiB, truncated to 20.00 KiB below, full version: https://github.com/llvm/llvm-project/pull/215611.diff
4 Files Affected:
- (modified) mlir/include/mlir/Dialect/Tosa/IR/TosaComplianceData.h.inc (+775-409)
- (modified) mlir/test/Dialect/Tosa/invalid_extension.mlir (+3-3)
- (modified) mlir/test/Dialect/Tosa/tosa-validation-version-1p0-invalid.mlir (+3-3)
- (modified) mlir/test/Dialect/Tosa/tosa-validation-version-1p1-valid.mlir (+1-1)
``````````diff
diff --git a/mlir/include/mlir/Dialect/Tosa/IR/TosaComplianceData.h.inc b/mlir/include/mlir/Dialect/Tosa/IR/TosaComplianceData.h.inc
index 628f68924e007..8b0eb071285cc 100644
--- a/mlir/include/mlir/Dialect/Tosa/IR/TosaComplianceData.h.inc
+++ b/mlir/include/mlir/Dialect/Tosa/IR/TosaComplianceData.h.inc
@@ -615,10 +615,16 @@ extensionComplianceMap = {
{{Extension::fp8e4m3},
{{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
SpecificationVersion::V_1_0},
+ {{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp16T, fp16T},
+ SpecificationVersion::V_1_1_DRAFT},
{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, fp32T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
+ {{fp8e4m3T, fp16T, fp16T, fp8e4m3T, fp16T, fp16T, fp16T},
+ SpecificationVersion::V_1_1_DRAFT},
{{fp8e4m3T, fp16T, fp16T, fp8e4m3T, fp16T, fp32T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
+ {{fp16T, fp8e4m3T, fp16T, fp16T, fp8e4m3T, fp16T, fp16T},
+ SpecificationVersion::V_1_1_DRAFT},
{{fp16T, fp8e4m3T, fp16T, fp16T, fp8e4m3T, fp32T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{fp8e4m3T, fp8e4m3T, fp32T, fp8e4m3T, fp8e4m3T, fp32T, fp32T},
@@ -634,10 +640,16 @@ extensionComplianceMap = {
{{Extension::fp8e5m2},
{{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
SpecificationVersion::V_1_0},
+ {{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp16T, fp16T},
+ SpecificationVersion::V_1_1_DRAFT},
{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, fp32T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
+ {{fp8e5m2T, fp16T, fp16T, fp8e5m2T, fp16T, fp16T, fp16T},
+ SpecificationVersion::V_1_1_DRAFT},
{{fp8e5m2T, fp16T, fp16T, fp8e5m2T, fp16T, fp32T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
+ {{fp16T, fp8e5m2T, fp16T, fp16T, fp8e5m2T, fp16T, fp16T},
+ SpecificationVersion::V_1_1_DRAFT},
{{fp16T, fp8e5m2T, fp16T, fp16T, fp8e5m2T, fp32T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{fp8e5m2T, fp8e5m2T, fp32T, fp8e5m2T, fp8e5m2T, fp32T, fp32T},
@@ -653,19 +665,9 @@ extensionComplianceMap = {
{{Extension::bf16},
{{{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
SpecificationVersion::V_1_0},
- {{fp16T, fp16T, fp16T, fp16T, fp16T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp16T, bf16T, fp16T, fp16T, bf16T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp16T, bf16T, fp16T, fp16T, bf16T, fp32T, fp16T},
+ {{bf16T, bf16T, bf16T, bf16T, bf16T, bf16T, bf16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, fp16T, fp16T, bf16T, fp16T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, fp16T, fp16T, bf16T, fp16T, fp32T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, bf16T, fp16T, bf16T, bf16T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, bf16T, fp16T, bf16T, bf16T, fp32T, fp16T},
+ {{bf16T, bf16T, bf16T, bf16T, bf16T, fp32T, bf16T},
SpecificationVersion::V_1_1_DRAFT},
{{fp16T, bf16T, fp32T, fp16T, bf16T, fp32T, fp32T},
SpecificationVersion::V_1_1_DRAFT},
@@ -677,34 +679,12 @@ extensionComplianceMap = {
SpecificationVersion::V_1_1_DRAFT},
{{fp32T, bf16T, fp32T, fp32T, bf16T, fp32T, fp32T},
SpecificationVersion::V_1_1_DRAFT}}},
- {{Extension::bf16, Extension::fp8e4m3},
- {{{fp8e4m3T, fp8e4m3T, fp16T, fp8e4m3T, fp8e4m3T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp8e4m3T, fp16T, fp16T, fp8e4m3T, fp16T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp8e4m3T, bf16T, fp16T, fp8e4m3T, bf16T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp8e4m3T, bf16T, fp16T, fp8e4m3T, bf16T, fp32T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp16T, fp8e4m3T, fp16T, fp16T, fp8e4m3T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, fp8e4m3T, fp16T, bf16T, fp8e4m3T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, fp8e4m3T, fp16T, bf16T, fp8e4m3T, fp32T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp8e4m3T, bf16T, fp32T, fp8e4m3T, bf16T, fp32T, fp32T},
+ {{Extension::fp8e4m3, Extension::fp8e5m2},
+ {{{fp8e4m3T, fp8e5m2T, fp16T, fp8e4m3T, fp8e5m2T, fp16T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, fp8e4m3T, fp32T, bf16T, fp8e4m3T, fp32T, fp32T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::bf16, Extension::fp8e4m3, Extension::fp8e5m2},
- {{{fp8e4m3T, fp8e5m2T, fp16T, fp8e4m3T, fp8e5m2T, bf16T, fp16T},
+ {{fp8e4m3T, fp8e5m2T, fp16T, fp8e4m3T, fp8e5m2T, fp32T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{fp8e5m2T, fp8e4m3T, fp16T, fp8e5m2T, fp8e4m3T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e4m3, Extension::fp8e5m2},
- {{{fp8e4m3T, fp8e5m2T, fp16T, fp8e4m3T, fp8e5m2T, fp32T, fp16T},
+ {{fp8e5m2T, fp8e4m3T, fp16T, fp8e5m2T, fp8e4m3T, fp16T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{fp8e5m2T, fp8e4m3T, fp16T, fp8e5m2T, fp8e4m3T, fp32T, fp16T},
SpecificationVersion::V_1_1_DRAFT},
@@ -713,17 +693,14 @@ extensionComplianceMap = {
{{fp8e5m2T, fp8e4m3T, fp32T, fp8e5m2T, fp8e4m3T, fp32T, fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
- Extension::mx_fp8e4m3},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e4m3T, fp32T, bf16T,
+ {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e4m3},
+ {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e4m3T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
+ {{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e4m3T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e4m3},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e4m3T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp8e4m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
@@ -736,17 +713,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
- Extension::mx_fp8e5m2},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e4m3T, fp32T, bf16T,
+ {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e5m2},
+ {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e4m3T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
+ {{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e4m3T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp8e5m2},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e4m3T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp8e5m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
@@ -759,17 +733,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
- Extension::mx_fp6e3m2},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e4m3T, fp32T, bf16T,
+ {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e3m2},
+ {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e4m3T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
+ {{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e4m3T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e3m2},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e4m3T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp6e3m2T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
@@ -782,17 +753,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
- Extension::mx_fp6e2m3},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e4m3T, fp32T, bf16T,
+ {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e2m3},
+ {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e4m3T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
+ {{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e4m3T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp6e2m3},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e4m3T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp6e2m3T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
@@ -805,17 +773,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
- Extension::mx_fp4e2m1},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e4m3T, fp32T, bf16T,
+ {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp4e2m1},
+ {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e4m3T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
+ {{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e4m3T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_fp4e2m1},
- {{{fp8e4m3T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e4m3T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp4e2m1T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
@@ -828,17 +793,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e4m3, Extension::mx_common,
- Extension::mx_int8},
- {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e4m3T, fp32T, bf16T,
+ {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_int8},
+ {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e4m3T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, bf16T,
+ {{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e4m3T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e4m3, Extension::mx_common, Extension::mx_int8},
- {{{fp8e4m3T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e4m3T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_mxint8T, fp8e4m3T, fp16T, fp32T, fp8e4m3T, fp32T,
@@ -851,37 +813,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e5m2},
- {{{fp8e5m2T, fp8e5m2T, fp16T, fp8e5m2T, fp8e5m2T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp8e5m2T, fp16T, fp16T, fp8e5m2T, fp16T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp8e5m2T, bf16T, fp16T, fp8e5m2T, bf16T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp8e5m2T, bf16T, fp16T, fp8e5m2T, bf16T, fp32T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp16T, fp8e5m2T, fp16T, fp16T, fp8e5m2T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, fp8e5m2T, fp16T, bf16T, fp8e5m2T, bf16T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, fp8e5m2T, fp16T, bf16T, fp8e5m2T, fp32T, fp16T},
- SpecificationVersion::V_1_1_DRAFT},
- {{fp8e5m2T, bf16T, fp32T, fp8e5m2T, bf16T, fp32T, fp32T},
- SpecificationVersion::V_1_1_DRAFT},
- {{bf16T, fp8e5m2T, fp32T, bf16T, fp8e5m2T, fp32T, fp32T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
- Extension::mx_fp8e4m3},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e5m2T, fp32T, bf16T,
+ {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e4m3},
+ {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e5m2T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
+ {{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e5m2T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e4m3},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp8e4m3T, fp16T, fp8e5m2T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp8e4m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
@@ -894,17 +833,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
- Extension::mx_fp8e5m2},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e5m2T, fp32T, bf16T,
+ {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e5m2},
+ {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e5m2T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
+ {{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e5m2T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp8e5m2},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp8e5m2T, fp16T, fp8e5m2T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp8e5m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
@@ -917,17 +853,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
- Extension::mx_fp6e3m2},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e5m2T, fp32T, bf16T,
+ {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e3m2},
+ {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e5m2T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
+ {{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e5m2T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e3m2},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp6e3m2T, fp16T, fp8e5m2T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp6e3m2T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
@@ -940,17 +873,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
- Extension::mx_fp6e2m3},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e5m2T, fp32T, bf16T,
+ {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e2m3},
+ {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e5m2T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
+ {{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e5m2T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp6e2m3},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp6e2m3T, fp16T, fp8e5m2T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp6e2m3T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
@@ -963,17 +893,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
- Extension::mx_fp4e2m1},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e5m2T, fp32T, bf16T,
+ {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp4e2m1},
+ {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e5m2T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
+ {{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e5m2T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_fp4e2m1},
- {{{fp8e5m2T, bs32_fp8ue8m0_fp4e2m1T, fp16T, fp8e5m2T, fp32T, fp32T,
+ SpecificationVersion::V_1_1_DRAFT},
+ {{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
{{bs32_fp8ue8m0_fp4e2m1T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, fp32T,
@@ -986,17 +913,14 @@ extensionComplianceMap = {
fp32T},
SpecificationVersion::V_1_1_DRAFT}},
allOf},
- {{Extension::bf16, Extension::fp8e5m2, Extension::mx_common,
- Extension::mx_int8},
- {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e5m2T, fp32T, bf16T,
+ {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_int8},
+ {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e5m2T, fp32T, fp16T,
fp16T},
SpecificationVersion::V_1_1_DRAFT},
- {{bs32_fp8ue8m0_mxint8T, fp8e5m2T, fp16T, fp32T, fp8e5m2T, bf16T,
+ {{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e5m2T, fp32T, fp32T,
fp16T},
- SpecificationVersion::V_1_1_DRAFT}},
- allOf},
- {{Extension::fp8e5m2, Extension::mx_common, Extension::mx_int8},
- {{{fp8e5m2T, bs32_fp8ue8m0_mxint8T, fp16T, fp8e5m2T, fp32T, fp32T,
+ SpecificationVersio...
[truncated]
``````````
</details>
https://github.com/llvm/llvm-project/pull/215611
More information about the Mlir-commits
mailing list