if (err)
                goto err_register_dpio_irq;
 
-       priv->io = dpaa2_io_create(&desc);
+       priv->io = dpaa2_io_create(&desc, dev);
        if (!priv->io) {
                dev_err(dev, "dpaa2_io_create failed\n");
                err = -ENOMEM;
 
        /* protect notifications list */
        spinlock_t lock_notifications;
        struct list_head notifications;
+       struct device *dev;
 };
 
 struct dpaa2_io_store {
 /**
  * dpaa2_io_create() - create a dpaa2_io object.
  * @desc: the dpaa2_io descriptor
+ * @dev: the actual DPIO device
  *
  * Activates a "struct dpaa2_io" corresponding to the given config of an actual
  * DPIO object.
  *
  * Return a valid dpaa2_io object for success, or NULL for failure.
  */
-struct dpaa2_io *dpaa2_io_create(const struct dpaa2_io_desc *desc)
+struct dpaa2_io *dpaa2_io_create(const struct dpaa2_io_desc *desc,
+                                struct device *dev)
 {
        struct dpaa2_io *obj = kmalloc(sizeof(*obj), GFP_KERNEL);
 
                dpio_by_cpu[desc->cpu] = obj;
        spin_unlock(&dpio_list_lock);
 
+       obj->dev = dev;
+
        return obj;
 }
 
 
        u32 qman_version;
 };
 
-struct dpaa2_io *dpaa2_io_create(const struct dpaa2_io_desc *desc);
+struct dpaa2_io *dpaa2_io_create(const struct dpaa2_io_desc *desc,
+                                struct device *dev);
 
 void dpaa2_io_down(struct dpaa2_io *d);