[Mlir-commits] [mlir] [mlir][x86vector] AVX512-BF16 Dot op (PR #124800)

[Adam Siemieniuk]

================
@@ -271,6 +271,94 @@ def Vp2IntersectQIntrOp : AVX512_IntrOp<"vp2intersect.q.512", 2, [
                    VectorOfLengthAndType<[8], [I64]>:$b);
 }
 
+//===----------------------------------------------------------------------===//
+// AVX512-BF16 op definitions
+//===----------------------------------------------------------------------===//
+
+// Operation that is part of the input dialect.
+class AVX512BF16_Op<string mnemonic, list<Trait> traits = []> :
+  Op<X86Vector_Dialect, "avx512bf16." # mnemonic, traits> {}
----------------
adam-smnk wrote:

Collapsed extensions into `.avx512.` within MLIR.
For LLVM intrinsic generation, `AVX512_IntrOp` now optionally accepts `extension` string that's used for LLVM intrinsic enum generation. 

https://github.com/llvm/llvm-project/pull/124800