xref: /freebsd/contrib/llvm-project/llvm/lib/Target/X86/X86InstrFMA3Info.h (revision 6966ac055c3b7a39266fb982493330df7a097997)
1 //===- X86InstrFMA3Info.h - X86 FMA3 Instruction Information ----*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file contains the implementation of the classes providing information
10 // about existing X86 FMA3 opcodes, classifying and grouping them.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_LIB_TARGET_X86_UTILS_X86INSTRFMA3INFO_H
15 #define LLVM_LIB_TARGET_X86_UTILS_X86INSTRFMA3INFO_H
16 
17 #include "X86.h"
18 #include "llvm/ADT/DenseMap.h"
19 #include <cassert>
20 #include <cstdint>
21 #include <set>
22 
23 namespace llvm {
24 
25 /// This class is used to group {132, 213, 231} forms of FMA opcodes together.
26 /// Each of the groups has either 3 opcodes, Also, each group has an attributes
27 /// field describing it.
28 struct X86InstrFMA3Group {
29   /// An array holding 3 forms of FMA opcodes.
30   uint16_t Opcodes[3];
31 
32   /// This bitfield specifies the attributes associated with the created
33   /// FMA groups of opcodes.
34   uint16_t Attributes;
35 
36   enum {
37     Form132,
38     Form213,
39     Form231,
40   };
41 
42   enum : uint16_t {
43     /// This bit must be set in the 'Attributes' field of FMA group if such
44     /// group of FMA opcodes consists of FMA intrinsic opcodes.
45     Intrinsic = 0x1,
46 
47     /// This bit must be set in the 'Attributes' field of FMA group if such
48     /// group of FMA opcodes consists of AVX512 opcodes accepting a k-mask and
49     /// passing the elements from the 1st operand to the result of the operation
50     /// when the correpondings bits in the k-mask are unset.
51     KMergeMasked = 0x2,
52 
53     /// This bit must be set in the 'Attributes' field of FMA group if such
54     /// group of FMA opcodes consists of AVX512 opcodes accepting a k-zeromask.
55     KZeroMasked = 0x4,
56   };
57 
58   /// Returns the 132 form of FMA opcode.
59   unsigned get132Opcode() const {
60     return Opcodes[Form132];
61   }
62 
63   /// Returns the 213 form of FMA opcode.
64   unsigned get213Opcode() const {
65     return Opcodes[Form213];
66   }
67 
68   /// Returns the 231 form of FMA opcode.
69   unsigned get231Opcode() const {
70     return Opcodes[Form231];
71   }
72 
73   /// Returns true iff the group of FMA opcodes holds intrinsic opcodes.
74   bool isIntrinsic() const { return (Attributes & Intrinsic) != 0; }
75 
76   /// Returns true iff the group of FMA opcodes holds k-merge-masked opcodes.
77   bool isKMergeMasked() const {
78     return (Attributes & KMergeMasked) != 0;
79   }
80 
81   /// Returns true iff the group of FMA opcodes holds k-zero-masked opcodes.
82   bool isKZeroMasked() const { return (Attributes &KZeroMasked) != 0; }
83 
84   /// Returns true iff the group of FMA opcodes holds any of k-masked opcodes.
85   bool isKMasked() const {
86     return (Attributes & (KMergeMasked | KZeroMasked)) != 0;
87   }
88 
89   bool operator<(const X86InstrFMA3Group &RHS) const {
90     return Opcodes[0] < RHS.Opcodes[0];
91   }
92 };
93 
94 /// Returns a reference to a group of FMA3 opcodes to where the given
95 /// \p Opcode is included. If the given \p Opcode is not recognized as FMA3
96 /// and not included into any FMA3 group, then nullptr is returned.
97 const X86InstrFMA3Group *getFMA3Group(unsigned Opcode, uint64_t TSFlags);
98 
99 } // end namespace llvm
100 
101 #endif // LLVM_LIB_TARGET_X86_UTILS_X86INSTRFMA3INFO_H
102