[llvm] [AMDGPU] Classify FLAT instructions as VMEM (PR #137148)

[Robert Imschweiler via llvm-commits]

================
@@ -2454,8 +2454,10 @@ bool SIInsertWaitcnts::isPreheaderToFlush(
 }
 
 bool SIInsertWaitcnts::isVMEMOrFlatVMEM(const MachineInstr &MI) const {
-  return SIInstrInfo::isVMEM(MI) ||
-         (SIInstrInfo::isFLAT(MI) && mayAccessVMEMThroughFlat(MI));
+  if (SIInstrInfo::isFLAT(MI))
+    return mayAccessVMEMThroughFlat(MI);
+  return SIInstrInfo::isMUBUF(MI) || SIInstrInfo::isMTBUF(MI) ||
+         SIInstrInfo::isImage(MI);
----------------
ro-i wrote:

Ah ok, maybe I was mistaken. I'd like to get it right - @arsenm what did you mean by "This should probably be something like if (isFLAT()) return mayAccessVMEMThroughFlat(); else // other non-flat cases"?

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