return 0;
 }
 
-static void s390_iommu_set_platform_dma(struct device *dev)
+/*
+ * Switch control over the IOMMU to S390's internal dma_api ops
+ */
+static int s390_iommu_platform_attach(struct iommu_domain *platform_domain,
+                                     struct device *dev)
 {
        struct zpci_dev *zdev = to_zpci_dev(dev);
 
+       if (!zdev->s390_domain)
+               return 0;
+
        __s390_iommu_detach_device(zdev);
        zpci_dma_init_device(zdev);
+       return 0;
 }
 
+static struct iommu_domain_ops s390_iommu_platform_ops = {
+       .attach_dev = s390_iommu_platform_attach,
+};
+
+static struct iommu_domain s390_iommu_platform_domain = {
+       .type = IOMMU_DOMAIN_PLATFORM,
+       .ops = &s390_iommu_platform_ops,
+};
+
 static void s390_iommu_get_resv_regions(struct device *dev,
                                        struct list_head *list)
 {
 }
 
 static const struct iommu_ops s390_iommu_ops = {
+       .default_domain = &s390_iommu_platform_domain,
        .capable = s390_iommu_capable,
        .domain_alloc = s390_domain_alloc,
        .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) {