uio: xilinx_ai_engine: Set the uio device as pdev driver data

It's needed when the driver is removed.

Signed-off-by: Hyun Kwon <hyun.kwon@xilinx.com>
Reviewed-by: Wendy Liang <wendy.liang@xilinx.com>
Signed-off-by: Michal Simek <michal.simek@xilinx.com>

diff --git a/drivers/uio/uio_xilinx_ai_engine.c b/drivers/uio/uio_xilinx_ai_engine.c
index 2a70b378b43bad3f1c802c4001864215c72e201b..69778d83c52fe4e5cb0666804ddb8bf61dd4bce3 100644 (file)
--- a/drivers/uio/uio_xilinx_ai_engine.c
+++ b/drivers/uio/uio_xilinx_ai_engine.c
@@ -133,6 +133,7 @@ static int xilinx_ai_engine_probe(struct platform_device *pdev)
        if (ret)
                goto err_out;
        platform_set_drvdata(uio, pdata);
+       platform_set_drvdata(pdev, uio);
 
        dev_info(&pdev->dev, "Xilinx AI Engine UIO driver probed");
        return 0;