[PATCH v2 3/3] arm: mach-k3: j784s4_init: Trigger LBIST and PBIST on MAIN R5 2_0

[Neha Malcom Francis]

Trigger all tests of PBIST and LBIST using appropriate calls to set the
core under test (MAIN R5 2_0) to it's required state.

Signed-off-by: Neha Malcom Francis <n-fran...@ti.com>
---
 arch/arm/mach-k3/j784s4/j784s4_init.c | 47 +++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/arch/arm/mach-k3/j784s4/j784s4_init.c 
b/arch/arm/mach-k3/j784s4/j784s4_init.c
index 787cf6261e4..f3df11f234f 100644
--- a/arch/arm/mach-k3/j784s4/j784s4_init.c
+++ b/arch/arm/mach-k3/j784s4/j784s4_init.c
@@ -17,6 +17,7 @@
 #include <dm/pinctrl.h>
 #include <mmc.h>
 #include <remoteproc.h>
+#include <k3_bist.h>
 
 #include "../sysfw-loader.h"
 #include "../common.h"
@@ -239,6 +240,8 @@ void k3_mem_init(void)
 void board_init_f(ulong dummy)
 {
        struct udevice *dev;
+       __maybe_unused struct bist_ops *ops;
+       __maybe_unused struct ti_sci_handle *handle;
        int ret;
 
        k3_spl_init();
@@ -251,6 +254,50 @@ void board_init_f(ulong dummy)
                        printf("AVS init failed: %d\n", ret);
        }
 
+       if (!IS_ENABLED(CONFIG_CPU_V7R) && IS_ENABLED(CONFIG_K3_BIST)) {
+               ret = uclass_get_device_by_driver(UCLASS_MISC,
+                                                 DM_DRIVER_GET(k3_bist),
+                                                 &dev);
+               if (ret)
+                       panic("Failed to get BIST device: %d\n", ret);
+
+               ops = (struct bist_ops *)device_get_ops(dev);
+               handle = get_ti_sci_handle();
+
+               /* get status of HW POST PBIST on MCU1_x */
+               ret = ops->run_pbist_post();
+               if (ret)
+                       panic("HW POST LBIST on MCU1_x failed: %d\n", ret);
+
+               /* trigger PBIST tests on MCU4_0 */
+               ret = prepare_pbist(handle);
+               ret |= ops->run_pbist_neg();
+               ret |= deprepare_pbist(handle);
+
+               ret |= prepare_pbist(handle);
+               ret |= ops->run_pbist();
+               ret |= deprepare_pbist(handle);
+
+               ret |= prepare_pbist(handle);
+               ret |= ops->run_pbist_rom();
+               ret |= deprepare_pbist(handle);
+
+               if (ret)
+                       panic("PBIST on MCU4_0 failed: %d\n", ret);
+
+               /* get status of HW POST PBIST on MCU1_x */
+               ret = ops->run_lbist_post();
+               if (ret)
+                       panic("HW POST LBIST on MCU1_x failed: %d\n", ret);
+
+               /* trigger LBIST tests on MCU1_x */
+               ret = prepare_lbist(handle);
+               ret |= ops->run_lbist();
+               ret |= deprepare_lbist(handle);
+               if (ret)
+                       panic("LBIST on MCU4_0 failed: %d\n", ret);
+       }
+
        if (IS_ENABLED(CONFIG_CPU_V7R))
                setup_navss_nb();
 
-- 
2.34.1