struct amdgpu_device *adev = (struct amdgpu_device *)handle;
        struct psp_context *psp = &adev->psp;
        int ret;
+       struct psp_runtime_boot_cfg_entry boot_cfg_entry;
 
        if (!amdgpu_sriov_vf(adev)) {
                ret = psp_init_microcode(psp);
                }
        }
 
+       memset(&boot_cfg_entry, 0, sizeof(boot_cfg_entry));
+       if (psp_get_runtime_db_entry(adev,
+                               PSP_RUNTIME_ENTRY_TYPE_BOOT_CONFIG,
+                               &boot_cfg_entry))
+               psp->boot_cfg_bitmask = boot_cfg_entry.boot_cfg_bitmask;
+
        ret = psp_memory_training_init(psp);
        if (ret) {
                DRM_ERROR("Failed to initialize memory training!\n");
 
        struct psp_securedisplay_context        securedisplay_context;
        struct mutex                    mutex;
        struct psp_memory_training_context mem_train_ctx;
+
+       uint32_t                        boot_cfg_bitmask;
 };
 
 struct amdgpu_psp_funcs {