1 files changed, 3 insertions, 0 deletions

diff --git a/drivers/iommu/amd_iommu.c b/drivers/iommu/amd_iommu.c
index 989c1ae03979..23c5be6a3a1e 100644
--- a/drivers/iommu/amd_iommu.c
+++ b/drivers/iommu/amd_iommu.c
@@ -1393,6 +1393,9 @@ static int iommu_map_page(struct protection_domain *dom,
         count     = PAGE_SIZE_PTE_COUNT(page_size);
         pte       = alloc_pte(dom, bus_addr, page_size, NULL, GFP_KERNEL);
 
+        if (!pte)
+                return -ENOMEM;
+
         for (i = 0; i < count; ++i)
                 if (IOMMU_PTE_PRESENT(pte[i]))
                         return -EBUSY;