1 files changed, 4 insertions, 9 deletions

diff --git a/drivers/gpu/drm/drm_cache.c b/drivers/gpu/drm/drm_cache.c
index a6b690626a6b..9a62d7a53553 100644
--- a/drivers/gpu/drm/drm_cache.c
+++ b/drivers/gpu/drm/drm_cache.c
@@ -32,6 +32,7 @@
 #include <drm/drmP.h>
 
 #if defined(CONFIG_X86)
+#include <asm/smp.h>
 
 /*
  * clflushopt is an unordered instruction which needs fencing with mfence or
@@ -64,12 +65,6 @@ static void drm_cache_flush_clflush(struct page *pages[],
                 drm_clflush_page(*pages++);
         mb();
 }
-
-static void
-drm_clflush_ipi_handler(void *null)
-{
-        wbinvd();
-}
 #endif
 
 void
@@ -82,7 +77,7 @@ drm_clflush_pages(struct page *pages[], unsigned long num_pages)
                 return;
         }
 
-        if (on_each_cpu(drm_clflush_ipi_handler, NULL, 1) != 0)
+        if (wbinvd_on_all_cpus())
                 printk(KERN_ERR "Timed out waiting for cache flush.\n");
 
 #elif defined(__powerpc__)
@@ -121,7 +116,7 @@ drm_clflush_sg(struct sg_table *st)
                 return;
         }
 
-        if (on_each_cpu(drm_clflush_ipi_handler, NULL, 1) != 0)
+        if (wbinvd_on_all_cpus())
                 printk(KERN_ERR "Timed out waiting for cache flush.\n");
 #else
         printk(KERN_ERR "Architecture has no drm_cache.c support\n");
@@ -144,7 +139,7 @@ drm_clflush_virt_range(void *addr, unsigned long length)
                 return;
         }
 
-        if (on_each_cpu(drm_clflush_ipi_handler, NULL, 1) != 0)
+        if (wbinvd_on_all_cpus())
                 printk(KERN_ERR "Timed out waiting for cache flush.\n");
 #else
         printk(KERN_ERR "Architecture has no drm_cache.c support\n");