const struct iommu_ops *iommu_ops_from_fwnode(const struct fwnode_handle *fwnode);
 
+static inline const struct iommu_ops *iommu_fwspec_ops(struct iommu_fwspec *fwspec)
+{
+       return iommu_ops_from_fwnode(fwspec ? fwspec->iommu_fwnode : NULL);
+}
+
 int iommu_group_replace_domain(struct iommu_group *group,
                               struct iommu_domain *new_domain);
 
 
 static int __iommu_probe_device(struct device *dev, struct list_head *group_list)
 {
        const struct iommu_ops *ops;
-       struct iommu_fwspec *fwspec;
        struct iommu_group *group;
        struct group_device *gdev;
        int ret;
         * be present, and that any of their registered instances has suitable
         * ops for probing, and thus cheekily co-opt the same mechanism.
         */
-       fwspec = dev_iommu_fwspec_get(dev);
-       if (fwspec && fwspec->ops)
-               ops = fwspec->ops;
-       else
-               ops = iommu_ops_from_fwnode(NULL);
-
+       ops = iommu_fwspec_ops(dev_iommu_fwspec_get(dev));
        if (!ops)
                return -ENODEV;
        /*
                return -EPROBE_DEFER;
 
        if (fwspec)
-               return ops == fwspec->ops ? 0 : -EINVAL;
+               return ops == iommu_fwspec_ops(fwspec) ? 0 : -EINVAL;
 
        if (!dev_iommu_get(dev))
                return -ENOMEM;
 
        fwnode_handle_get(iommu_fwnode);
        fwspec->iommu_fwnode = iommu_fwnode;
-       fwspec->ops = ops;
        dev_iommu_fwspec_set(dev, fwspec);
        return 0;
 }
 
 #include <linux/slab.h>
 #include <linux/fsl/mc.h>
 
+#include "iommu-priv.h"
+
 static int of_iommu_xlate(struct device *dev,
                          struct of_phandle_args *iommu_spec)
 {
        if (ret)
                return ret;
 
-       ops = dev_iommu_fwspec_get(dev)->ops;
+       ops = iommu_ops_from_fwnode(&iommu_spec->np->fwnode);
        if (!ops->of_xlate || !try_module_get(ops->owner))
                return -ENODEV;
 
 
 
 /**
  * struct iommu_fwspec - per-device IOMMU instance data
- * @ops: ops for this device's IOMMU
  * @iommu_fwnode: firmware handle for this device's IOMMU
  * @flags: IOMMU_FWSPEC_* flags
  * @num_ids: number of associated device IDs
  * consumers.
  */
 struct iommu_fwspec {
-       const struct iommu_ops  *ops;
        struct fwnode_handle    *iommu_fwnode;
        u32                     flags;
        unsigned int            num_ids;