[DirectX][ShaderFlags] Add analysis for WaveOps flag

- Add emitDXILOpProperties to build boolean tables of which ops hold
which properties into `DXILEmitter.cpp`
- Check each call instruction to see if it invokes an intrinsic that is
marked as IsWave, and set the WaveOps flag if this is true for any
intrinsics. Done in DXILShaderFlags.cpp

Author: inbelic

llvm/lib/Target/DirectX/DXILShaderFlags.cpp

@@ -11,6 +11,7 @@
 ///
 //===----------------------------------------------------------------------===//
 
+#include "DXILConstants.h"
 #include "DXILShaderFlags.h"
 #include "DirectX.h"
 #include "llvm/ADT/SCCIterator.h"
@@ -30,6 +31,25 @@
 using namespace llvm;
 using namespace llvm::dxil;
 
+static dxil::Properties getOpCodeProperties(dxil::OpCode OpCode) {
+  dxil::Properties Props;
+  switch (OpCode) {
+#define DXIL_OP_PROPERTIES(OpCode, ...) \
+  case OpCode: Props = dxil::Properties{__VA_ARGS__}; break;
+#include "DXILOperation.inc"
+  }
+  return Props;
+}
+
+static bool checkWaveOps(Intrinsic::ID IID) {
+  switch (IID) {
+#define DXIL_OP_INTRINSIC(OpCode, IntrinsicID, ...) \
+  case IntrinsicID: return getOpCodeProperties(OpCode).IsWave;
+#include "DXILOperation.inc"
+  }
+  return false;
+}
+
 /// Update the shader flags mask based on the given instruction.
 /// \param CSF Shader flags mask to update.
 /// \param I Instruction to check.
@@ -92,6 +112,8 @@ void ModuleShaderFlags::updateFunctionFlags(ComputedShaderFlags &CSF,
 
     // TODO: Set DX11_1_DoubleExtensions if I is a call to DXIL intrinsic
     // DXIL::Opcode::Fma https://github.com/llvm/llvm-project/issues/114554
+
+    CSF.WaveOps |= checkWaveOps(CI->getIntrinsicID());
   }
 }
 

llvm/utils/TableGen/DXILEmitter.cpp

@@ -437,6 +437,21 @@ static void emitDXILProperties(const RecordKeeper &Records, raw_ostream &OS) {
   OS << "#endif\n\n";
 }
 
+/// Emit a table of bools denoting a DXIL op's properties
+static void emitDXILOpProperties(const RecordKeeper &Records,
+                                 ArrayRef<DXILOperationDesc> Ops,
+                                 raw_ostream &OS) {
+  auto DefinedProps = Records.getAllDerivedDefinitions("DXILProperty");
+  OS << "#ifdef DXIL_OP_PROPERTIES\n";
+  for (const auto &Op : Ops) {
+    OS << "DXIL_OP_PROPERTIES(dxil::OpCode::" << Op.OpName;
+    emitBoolTable(Op.PropRecs, DefinedProps, OS);
+    OS << ")\n";
+  }
+  OS << "#undef DXIL_OP_PROPERTIES\n";
+  OS << "#endif\n\n";
+}
+
 /// Emit a list of DXIL op function types
 static void emitDXILOpFunctionTypes(ArrayRef<DXILOperationDesc> Ops,
                                     raw_ostream &OS) {
@@ -641,6 +656,7 @@ static void emitDxilOperation(const RecordKeeper &Records, raw_ostream &OS) {
   emitDXILAttributes(Records, OS);
   emitDXILOpAttributes(Records, DXILOps, OS);
   emitDXILProperties(Records, OS);
+  emitDXILOpProperties(Records, DXILOps, OS);
   emitDXILOpFunctionTypes(DXILOps, OS);
   emitDXILIntrinsicArgSelectTypes(Records, OS);
   emitDXILIntrinsicMap(DXILOps, OS);