return ret;
 }
 
-static void fsl_pamu_detach_device(struct iommu_domain *domain,
-                                  struct device *dev)
+static void fsl_pamu_set_platform_dma(struct device *dev)
 {
+       struct iommu_domain *domain = iommu_get_domain_for_dev(dev);
        struct fsl_dma_domain *dma_domain = to_fsl_dma_domain(domain);
        const u32 *prop;
        int len;
        .domain_alloc   = fsl_pamu_domain_alloc,
        .probe_device   = fsl_pamu_probe_device,
        .device_group   = fsl_pamu_device_group,
+       .set_platform_dma_ops = fsl_pamu_set_platform_dma;
        .default_domain_ops = &(const struct iommu_domain_ops) {
                .attach_dev     = fsl_pamu_attach_device,
-               .detach_dev     = fsl_pamu_detach_device,
                .iova_to_phys   = fsl_pamu_iova_to_phys,
                .free           = fsl_pamu_domain_free,
        }
 
        return ret;
 }
 
-static void msm_iommu_detach_dev(struct iommu_domain *domain,
-                                struct device *dev)
+static void msm_iommu_set_platform_dma(struct device *dev)
 {
+       struct iommu_domain *domain = iommu_get_domain_for_dev(dev);
        struct msm_priv *priv = to_msm_priv(domain);
        unsigned long flags;
        struct msm_iommu_dev *iommu;
        .domain_alloc = msm_iommu_domain_alloc,
        .probe_device = msm_iommu_probe_device,
        .device_group = generic_device_group,
+       .set_platform_dma_ops = msm_iommu_set_platform_dma,
        .pgsize_bitmap = MSM_IOMMU_PGSIZES,
        .of_xlate = qcom_iommu_of_xlate,
        .default_domain_ops = &(const struct iommu_domain_ops) {
                .attach_dev     = msm_iommu_attach_dev,
-               .detach_dev     = msm_iommu_detach_dev,
                .map_pages      = msm_iommu_map,
                .unmap_pages    = msm_iommu_unmap,
                /*
 
        return 0;
 }
 
-static void mtk_iommu_v1_detach_device(struct iommu_domain *domain, struct device *dev)
+static void mtk_iommu_v1_set_platform_dma(struct device *dev)
 {
        struct mtk_iommu_v1_data *data = dev_iommu_priv_get(dev);
 
        .def_domain_type = mtk_iommu_v1_def_domain_type,
        .device_group   = generic_device_group,
        .pgsize_bitmap  = MT2701_IOMMU_PAGE_SIZE,
+       .set_platform_dma_ops = mtk_iommu_v1_set_platform_dma,
        .owner          = THIS_MODULE,
        .default_domain_ops = &(const struct iommu_domain_ops) {
                .attach_dev     = mtk_iommu_v1_attach_device,
-               .detach_dev     = mtk_iommu_v1_detach_device,
                .map_pages      = mtk_iommu_v1_map,
                .unmap_pages    = mtk_iommu_v1_unmap,
                .iova_to_phys   = mtk_iommu_v1_iova_to_phys,
 
        omap_domain->dev = NULL;
 }
 
-static void omap_iommu_detach_dev(struct iommu_domain *domain,
-                                 struct device *dev)
+static void omap_iommu_set_platform_dma(struct device *dev)
 {
+       struct iommu_domain *domain = iommu_get_domain_for_dev(dev);
        struct omap_iommu_domain *omap_domain = to_omap_domain(domain);
 
        spin_lock(&omap_domain->lock);
        .probe_device   = omap_iommu_probe_device,
        .release_device = omap_iommu_release_device,
        .device_group   = omap_iommu_device_group,
+       .set_platform_dma_ops = omap_iommu_set_platform_dma,
        .pgsize_bitmap  = OMAP_IOMMU_PGSIZES,
        .default_domain_ops = &(const struct iommu_domain_ops) {
                .attach_dev     = omap_iommu_attach_dev,
-               .detach_dev     = omap_iommu_detach_dev,
                .map            = omap_iommu_map,
                .unmap          = omap_iommu_unmap,
                .iova_to_phys   = omap_iommu_iova_to_phys,
 
        return 0;
 }
 
-static void s390_iommu_detach_device(struct iommu_domain *domain,
-                                    struct device *dev)
+static void s390_iommu_set_platform_dma(struct device *dev)
 {
        struct zpci_dev *zdev = to_zpci_dev(dev);
 
-       WARN_ON(zdev->s390_domain != to_s390_domain(domain));
-
        __s390_iommu_detach_device(zdev);
        zpci_dma_init_device(zdev);
 }
        .probe_device = s390_iommu_probe_device,
        .release_device = s390_iommu_release_device,
        .device_group = generic_device_group,
+       .set_platform_dma_ops = s390_iommu_set_platform_dma,
        .pgsize_bitmap = SZ_4K,
        .get_resv_regions = s390_iommu_get_resv_regions,
        .default_domain_ops = &(const struct iommu_domain_ops) {
                .attach_dev     = s390_iommu_attach_device,
-               .detach_dev     = s390_iommu_detach_device,
                .map_pages      = s390_iommu_map_pages,
                .unmap_pages    = s390_iommu_unmap_pages,
                .flush_iotlb_all = s390_iommu_flush_iotlb_all,
 
        return ret;
 }
 
-static void gart_iommu_detach_dev(struct iommu_domain *domain,
-                                 struct device *dev)
+static void gart_iommu_set_platform_dma(struct device *dev)
 {
+       struct iommu_domain *domain = iommu_get_domain_for_dev(dev);
        struct gart_device *gart = gart_handle;
 
        spin_lock(&gart->dom_lock);
        .domain_alloc   = gart_iommu_domain_alloc,
        .probe_device   = gart_iommu_probe_device,
        .device_group   = generic_device_group,
+       .set_platform_dma_ops = gart_iommu_set_platform_dma,
        .pgsize_bitmap  = GART_IOMMU_PGSIZES,
        .of_xlate       = gart_iommu_of_xlate,
        .default_domain_ops = &(const struct iommu_domain_ops) {
                .attach_dev     = gart_iommu_attach_dev,
-               .detach_dev     = gart_iommu_detach_dev,
                .map            = gart_iommu_map,
                .unmap          = gart_iommu_unmap,
                .iova_to_phys   = gart_iommu_iova_to_phys,
 
        return err;
 }
 
-static void tegra_smmu_detach_dev(struct iommu_domain *domain, struct device *dev)
+static void tegra_smmu_set_platform_dma(struct device *dev)
 {
+       struct iommu_domain *domain = iommu_get_domain_for_dev(dev);
        struct iommu_fwspec *fwspec = dev_iommu_fwspec_get(dev);
        struct tegra_smmu_as *as = to_smmu_as(domain);
        struct tegra_smmu *smmu = as->smmu;
        .domain_alloc = tegra_smmu_domain_alloc,
        .probe_device = tegra_smmu_probe_device,
        .device_group = tegra_smmu_device_group,
+       .set_platform_dma_ops = tegra_smmu_set_platform_dma,
        .of_xlate = tegra_smmu_of_xlate,
        .pgsize_bitmap = SZ_4K,
        .default_domain_ops = &(const struct iommu_domain_ops) {
                .attach_dev     = tegra_smmu_attach_dev,
-               .detach_dev     = tegra_smmu_detach_dev,
                .map            = tegra_smmu_map,
                .unmap          = tegra_smmu_unmap,
                .iova_to_phys   = tegra_smmu_iova_to_phys,