From ddc736009c0941535c6d498eb43c673dd6051961 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Deronne?= <sebastien.deronne@gmail.com>
Date: Tue, 4 Jul 2023 12:08:57 -0400
Subject: [PATCH] wifi: Extend use of EHT PPDU type for MU-MIMO

---
 src/wifi/model/wifi-tx-vector.cc | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/wifi/model/wifi-tx-vector.cc b/src/wifi/model/wifi-tx-vector.cc
index 72845e961..8709ceae4 100644
--- a/src/wifi/model/wifi-tx-vector.cc
+++ b/src/wifi/model/wifi-tx-vector.cc
@@ -529,6 +529,10 @@ WifiTxVector::IsDlOfdma() const
     {
         return false;
     }
+    if (IsEht(m_preamble))
+    {
+        return m_ehtPpduType == 0;
+    }
     if (m_muUserInfos.size() == 1)
     {
         return true;
@@ -552,6 +556,10 @@ WifiTxVector::IsDlMuMimo() const
     {
         return false;
     }
+    if (IsEht(m_preamble))
+    {
+        return m_ehtPpduType == 2;
+    }
     if (m_muUserInfos.size() < 2)
     {
         return false;