return;
 
        /* Check if it is SEV command completion: */
-       reg = ioread32(sev->io_regs + sev->psp->vdata->cmdresp_reg);
+       reg = ioread32(sev->io_regs + sev->vdata->cmdresp_reg);
        if (reg & PSP_CMDRESP_RESP) {
                sev->int_rcvd = 1;
                wake_up(&sev->int_queue);
        if (!ret)
                return -ETIMEDOUT;
 
-       *reg = ioread32(sev->io_regs + sev->psp->vdata->cmdresp_reg);
+       *reg = ioread32(sev->io_regs + sev->vdata->cmdresp_reg);
 
        return 0;
 }
        print_hex_dump_debug("(in):  ", DUMP_PREFIX_OFFSET, 16, 2, data,
                             sev_cmd_buffer_len(cmd), false);
 
-       iowrite32(phys_lsb, sev->io_regs + psp->vdata->cmdbuff_addr_lo_reg);
-       iowrite32(phys_msb, sev->io_regs + psp->vdata->cmdbuff_addr_hi_reg);
+       iowrite32(phys_lsb, sev->io_regs + sev->vdata->cmdbuff_addr_lo_reg);
+       iowrite32(phys_msb, sev->io_regs + sev->vdata->cmdbuff_addr_hi_reg);
 
        sev->int_rcvd = 0;
 
        reg = cmd;
        reg <<= SEV_CMDRESP_CMD_SHIFT;
        reg |= SEV_CMDRESP_IOC;
-       iowrite32(reg, sev->io_regs + psp->vdata->cmdresp_reg);
+       iowrite32(reg, sev->io_regs + sev->vdata->cmdresp_reg);
 
        /* wait for command completion */
        ret = sev_wait_cmd_ioc(sev, ®, psp_timeout);
 
        sev->io_regs = psp->io_regs;
 
+       sev->vdata = (struct sev_vdata *)psp->vdata->sev;
+       if (!sev->vdata) {
+               ret = -ENODEV;
+               dev_err(dev, "sev: missing driver data\n");
+               goto e_err;
+       }
+
        psp_set_sev_irq_handler(psp, sev_irq_handler, sev);
 
        ret = sev_misc_init(sev);
 
        const unsigned int rsamax;
 };
 
-struct psp_vdata {
+struct sev_vdata {
        const unsigned int cmdresp_reg;
        const unsigned int cmdbuff_addr_lo_reg;
        const unsigned int cmdbuff_addr_hi_reg;
+};
+
+struct psp_vdata {
+       const struct sev_vdata *sev;
        const unsigned int feature_reg;
        const unsigned int inten_reg;
        const unsigned int intsts_reg;
 
 #endif
 
 #ifdef CONFIG_CRYPTO_DEV_SP_PSP
-static const struct psp_vdata pspv1 = {
+static const struct sev_vdata sevv1 = {
        .cmdresp_reg            = 0x10580,
        .cmdbuff_addr_lo_reg    = 0x105e0,
        .cmdbuff_addr_hi_reg    = 0x105e4,
+};
+
+static const struct sev_vdata sevv2 = {
+       .cmdresp_reg            = 0x10980,
+       .cmdbuff_addr_lo_reg    = 0x109e0,
+       .cmdbuff_addr_hi_reg    = 0x109e4,
+};
+
+static const struct psp_vdata pspv1 = {
+       .sev                    = &sevv1,
        .feature_reg            = 0x105fc,
        .inten_reg              = 0x10610,
        .intsts_reg             = 0x10614,
 };
 
 static const struct psp_vdata pspv2 = {
-       .cmdresp_reg            = 0x10980,
-       .cmdbuff_addr_lo_reg    = 0x109e0,
-       .cmdbuff_addr_hi_reg    = 0x109e4,
+       .sev                    = &sevv2,
        .feature_reg            = 0x109fc,
        .inten_reg              = 0x10690,
        .intsts_reg             = 0x10694,