Changed based on review feedback

Minor change

Author: jgu222

visa/G4_SendDescs.cpp

@@ -1745,6 +1745,35 @@ bool G4_SendDescRaw::isBarrier() const {
   return funcID == SFID::GATEWAY && (funcCtrl & 0xFF) == 0x4;
 }
 
+bool G4_SendDescRaw::isBTS() const {
+  if (isLscOp()) {
+    switch (getLscAddrType()) {
+    case LSC_ADDR_TYPE_BSS:
+    case LSC_ADDR_TYPE_SS:
+    case LSC_ADDR_TYPE_BTI:
+      return true;
+    default:
+      break;
+    }
+  } else {
+    const G4_Operand *BTI = getBti();
+    uint32_t BTIImm = 0;
+    if (BTI && !BTI->isImm()) {
+      // Desc in reg, should be stateful.
+      return true;
+    } else if (BTI) {
+      BTIImm = (BTI->asImm()->getInt() & 0xFF);
+    } else {
+      BTIImm = (getDesc() & 0xFF);
+    }
+    constexpr uint32_t BSS_BTI = 252;
+    constexpr uint32_t BTI_MAX = 240;
+    if (BTIImm <= BTI_MAX || BTIImm == BSS_BTI)
+      return true;
+  }
+  return false;
+}
+
 std::optional<ImmOff> G4_SendDescRaw::getOffset() const {
   if (isLscOp()) {
     // technically unavailable until XE2, but this is binary compatible,

visa/G4_SendDescs.hpp

@@ -297,6 +297,7 @@ class G4_SendDesc {
   bool isGTWY() const {return getSFID() == SFID::GATEWAY;}
   //
   virtual bool isSLM() const = 0;
+  virtual bool isBTS() const = 0; // BTS stateful
   virtual bool isTyped() const = 0;
   virtual bool isAtomic() const = 0;
   virtual bool isBarrier() const = 0;
@@ -680,6 +681,7 @@ class G4_SendDescRaw : public G4_SendDesc {
   virtual std::optional<ImmOff> getOffset() const override;
 
   virtual bool isSLM() const override { return isSLMMessage(); }
+  virtual bool isBTS() const override;
   virtual bool isAtomic() const override { return isAtomicMessage(); }
   virtual bool isBarrier() const override;
   virtual bool isFence() const override;