mpm: Add x4xx DRAM BIST

1 files changed, 23 insertions, 1 deletions

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