[Mlir-commits] [mlir] [mlir][x86vector] AVX512-BF16 Dot op (PR #124800)
Adam Siemieniuk
llvmlistbot at llvm.org
Tue Jan 28 11:28:35 PST 2025
================
@@ -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:
AFAIK, most intrinsics already follow naming convention of `x86_avx512_`. However, for `fp16` and `bf16` types, intrinsics have the type explicitly as a part of the name `x86_avx512(fp16|bf16)_`.
This new definition follows the current convention and mirrors the explicit naming style.
I think we could generalize it to have these ops under single `x86vector.avx512.` but honestly either way seems fine to me.
https://github.com/llvm/llvm-project/pull/124800
More information about the Mlir-commits
mailing list