static int domain_context_mapping_one(struct dmar_domain *domain,
                                      struct intel_iommu *iommu,
+                                     struct pasid_table *table,
                                      u8 bus, u8 devfn)
 {
        u16 did = domain->iommu_did[iommu->seq_id];
 struct domain_context_mapping_data {
        struct dmar_domain *domain;
        struct intel_iommu *iommu;
+       struct pasid_table *table;
 };
 
 static int domain_context_mapping_cb(struct pci_dev *pdev,
        struct domain_context_mapping_data *data = opaque;
 
        return domain_context_mapping_one(data->domain, data->iommu,
-                                         PCI_BUS_NUM(alias), alias & 0xff);
+                                         data->table, PCI_BUS_NUM(alias),
+                                         alias & 0xff);
 }
 
 static int
 domain_context_mapping(struct dmar_domain *domain, struct device *dev)
 {
+       struct domain_context_mapping_data data;
+       struct pasid_table *table;
        struct intel_iommu *iommu;
        u8 bus, devfn;
-       struct domain_context_mapping_data data;
 
        iommu = device_to_iommu(dev, &bus, &devfn);
        if (!iommu)
                return -ENODEV;
 
+       table = intel_pasid_get_table(dev);
+
        if (!dev_is_pci(dev))
-               return domain_context_mapping_one(domain, iommu, bus, devfn);
+               return domain_context_mapping_one(domain, iommu, table,
+                                                 bus, devfn);
 
        data.domain = domain;
        data.iommu = iommu;
+       data.table = table;
 
        return pci_for_each_dma_alias(to_pci_dev(dev),
                                      &domain_context_mapping_cb, &data);