dm: spi: Add support for all targets which requires MANUAL_RELOC

It is follow up patch based on
"dm: Add support for all targets which requires MANUAL_RELOC"
(sha1: 484fdf5ba058b07be5ca82763aa2b72063540ef3)
to update function pointers for DM.

Signed-off-by: Michal Simek <michal.simek@xilinx.com>
Acked-by: Simon Glass <sjg@chromium.org>

diff --git a/drivers/spi/spi-uclass.c b/drivers/spi/spi-uclass.c
index 58388ef..3c7d64a 100644 (file)
--- a/drivers/spi/spi-uclass.c
+++ b/drivers/spi/spi-uclass.c
@@ -118,6 +118,26 @@ static int spi_post_probe(struct udevice *bus)
        spi->max_hz = fdtdec_get_int(gd->fdt_blob, bus->of_offset,
                                     "spi-max-frequency", 0);
 
+#if defined(CONFIG_NEEDS_MANUAL_RELOC)
+       struct dm_spi_ops *ops = spi_get_ops(bus);
+
+
+       if (ops->claim_bus)
+               ops->claim_bus += gd->reloc_off;
+       if (ops->release_bus)
+               ops->release_bus += gd->reloc_off;
+       if (ops->set_wordlen)
+               ops->set_wordlen += gd->reloc_off;
+       if (ops->xfer)
+               ops->xfer += gd->reloc_off;
+       if (ops->set_speed)
+               ops->set_speed += gd->reloc_off;
+       if (ops->set_mode)
+               ops->set_mode += gd->reloc_off;
+       if (ops->cs_info)
+               ops->cs_info += gd->reloc_off;
+#endif
+
        return 0;
 }