1 files changed, 13 insertions, 9 deletions

diff --git a/arch/sparc64/kernel/pci_iommu.c b/arch/sparc64/kernel/pci_iommu.c
index 66712772f49..9e405cbbcb0 100644
--- a/arch/sparc64/kernel/pci_iommu.c
+++ b/arch/sparc64/kernel/pci_iommu.c
@@ -37,17 +37,21 @@
 /* Must be invoked under the IOMMU lock. */
 static void __iommu_flushall(struct iommu *iommu)
 {
-        unsigned long tag;
-        int entry;
+        if (iommu->iommu_flushinv) {
+                pci_iommu_write(iommu->iommu_flushinv, ~(u64)0);
+        } else {
+                unsigned long tag;
+                int entry;
 
-        tag = iommu->iommu_flush + (0xa580UL - 0x0210UL);
-        for (entry = 0; entry < 16; entry++) {
-                pci_iommu_write(tag, 0);
-                tag += 8;
-        }
+                tag = iommu->iommu_flush + (0xa580UL - 0x0210UL);
+                for (entry = 0; entry < 16; entry++) {
+                        pci_iommu_write(tag, 0);
+                        tag += 8;
+                }
 
-        /* Ensure completion of previous PIO writes. */
-        (void) pci_iommu_read(iommu->write_complete_reg);
+                /* Ensure completion of previous PIO writes. */
+                (void) pci_iommu_read(iommu->write_complete_reg);
+        }
 }
 
 #define IOPTE_CONSISTENT(CTX) \