{
        struct usnic_ib_pd *pd = to_upd(ibpd);
 
-       pd->umem_pd = usnic_uiom_alloc_pd();
+       pd->umem_pd = usnic_uiom_alloc_pd(ibpd->device->dev.parent);
        if (IS_ERR(pd->umem_pd))
                return PTR_ERR(pd->umem_pd);
 
        usnic_err("No VF %u found\n", vfid);
        return -EINVAL;
 }
-
 
 #include <linux/iommu.h>
 #include <linux/workqueue.h>
 #include <linux/list.h>
-#include <linux/pci.h>
 #include <rdma/ib_verbs.h>
 
 #include "usnic_log.h"
        __usnic_uiom_release_tail(uiomr);
 }
 
-struct usnic_uiom_pd *usnic_uiom_alloc_pd(void)
+struct usnic_uiom_pd *usnic_uiom_alloc_pd(struct device *dev)
 {
        struct usnic_uiom_pd *pd;
        void *domain;
        if (!pd)
                return ERR_PTR(-ENOMEM);
 
-       pd->domain = domain = iommu_domain_alloc(&pci_bus_type);
+       pd->domain = domain = iommu_domain_alloc(dev->bus);
        if (!domain) {
                usnic_err("Failed to allocate IOMMU domain");
                kfree(pd);
 
        struct scatterlist              page_list[];
 };
 
-struct usnic_uiom_pd *usnic_uiom_alloc_pd(void);
+struct usnic_uiom_pd *usnic_uiom_alloc_pd(struct device *dev);
 void usnic_uiom_dealloc_pd(struct usnic_uiom_pd *pd);
 int usnic_uiom_attach_dev_to_pd(struct usnic_uiom_pd *pd, struct device *dev);
 void usnic_uiom_detach_dev_from_pd(struct usnic_uiom_pd *pd,