From f748b60d9033441e62531bf1f0790747e8f3d29c Mon Sep 17 00:00:00 2001
From: Wade Fife <wade.fife@ettus.com>
Date: Wed, 26 Jan 2022 15:19:41 -0600
Subject: mpm: Add x4xx DRAM BIST

---
 mpm/python/x4xx_bist | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/mpm/python/x4xx_bist b/mpm/python/x4xx_bist
index 0c6fb1155..60c1799c5 100644
--- a/mpm/python/x4xx_bist
+++ b/mpm/python/x4xx_bist
@@ -79,7 +79,7 @@ class X4XXBIST(bist.UsrpBIST):
     # Default FPGA image type
     DEFAULT_FPGA_TYPE = 'X4_200'
     lv_compat_format = {
-        'ddr3': {
+        'dram': {
             'throughput': -1,
         },
         'gpsdo': {
@@ -211,6 +211,28 @@ class X4XXBIST(bist.UsrpBIST):
 # BISTS
 # All bist_* methods must return True/False success values!
 #############################################################################
+    def bist_dram(self):
+        """
+        BIST for PL DDR4 DRAM
+        Description: Calls a test to examine the speed of the DRAM. To be
+        precise, it fires up a UHD session, which runs a DRAM BiST internally.
+        If that works, it'll return estimated throughput that was gathered
+        during the DRAM BiST.
+
+        External Equipment: None
+
+        Return dictionary:
+        - throughput: The estimated throughput in bytes/s
+
+        Return status:
+        True if the DRAM bist passed
+        """
+        assert 'dram' in self.tests_to_run
+        if self.args.dry_run:
+            return True, {'throughput': 1700e6}
+        result = bist.test_ddr3_with_usrp_probe()
+        return result.get('throughput', 0) > 1500e6, result
+
     def bist_gpsdo(self):
         """
         BIST for GPSDO
-- 
cgit v1.2.3-59-g8ed1b