[DXIL][ShaderFlags] Add analysis for WaveOps flag

- updates DXILShaderFlags to check if any functions call a wave
intrinsic
- adds a testcase to iswave.ll

Author: inbelic

llvm/lib/Target/DirectX/DXILShaderFlags.cpp

@@ -12,14 +12,32 @@
 //===----------------------------------------------------------------------===//
 
 #include "DXILShaderFlags.h"
+#include "DXILConstants.h"
 #include "DirectX.h"
 #include "llvm/IR/Instruction.h"
+#include "llvm/IR/Instructions.h"
+#include "llvm/IR/IntrinsicsDirectX.h"
 #include "llvm/IR/Module.h"
 #include "llvm/Support/FormatVariadic.h"
 
 using namespace llvm;
 using namespace llvm::dxil;
 
+// Include hasProperty which is generated by tabelGen
+#define DXIL_OP_PROPERTY_HELPER
+#include "DXILOperation.inc"
+#undef DXIL_OP_PROPERTY_HELPER
+
+static bool checkWaveOps(const CallInst *CI) {
+  switch (CI->getIntrinsicID()) {
+#define DXIL_OP_INTRINSIC(OpCode, Intrin)                                      \
+  case Intrin:                                                                 \
+    return hasProperty(OpCode, dxil::Property::IsWave);
+#include "DXILOperation.inc"
+  }
+  return false;
+}
+
 static void updateFlags(ComputedShaderFlags &Flags, const Instruction &I) {
   Type *Ty = I.getType();
   if (Ty->isDoubleTy()) {
@@ -34,6 +52,9 @@ static void updateFlags(ComputedShaderFlags &Flags, const Instruction &I) {
       break;
     }
   }
+
+  if (auto *CI = dyn_cast<CallInst>(&I))
+    Flags.WaveOps |= checkWaveOps(CI);
 }
 
 ComputedShaderFlags ComputedShaderFlags::computeFlags(Module &M) {

llvm/test/CodeGen/DirectX/ShaderFlags/iswave.ll

@@ -0,0 +1,37 @@
+; RUN: opt -S --passes="print-dx-shader-flags" 2>&1 %s | FileCheck %s
+
+target triple = "dxil-pc-shadermodel6.7-library"
+
+; CHECK: ; Shader Flags Value: 0x00080000
+; CHECK: ; Note: shader requires additional functionality:
+; CHECK-NEXT: ;       Wave level operations
+; CHECK-NEXT: ; Note: extra DXIL module flags:
+; CHECK-NEXT: {{^;$}}
+
+; TODO: is there a way that we can split the input file into many?
+; as the shader flag will be true if any are set
+
+define i32 @wrla(i32 %expr, i32 %idx) {
+  %ret = call i32 @llvm.dx.wave.readlane.i32(i32 %expr, i32 %idx)
+  ret i32 %ret
+}
+
+define i1 @waat(i1 %x) {
+  %ret = call i1 @llvm.dx.wave.any(i1 %x)
+  ret i1 %ret
+}
+
+define i32 @wabc(i1 %x) {
+  %ret = call i32 @llvm.dx.wave.active.countbits(i1 %x)
+  ret i32 %ret
+}
+
+define i1 @wifl() {
+  %ret = call i1 @llvm.dx.wave.is.first.lane()
+  ret i1 %ret
+}
+
+define i32 @wgli() {
+  %ret = call i32 @llvm.dx.wave.getlaneindex()
+  ret i32 %ret
+}