#include <drm/drmP.h>
 
 #if defined(CONFIG_X86)
+#include <asm/smp.h>
 
 /*
  * clflushopt is an unordered instruction which needs fencing with mfence or
                drm_clflush_page(*pages++);
        mb();
 }
-
-static void
-drm_clflush_ipi_handler(void *null)
-{
-       wbinvd();
-}
 #endif
 
 void
                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__)
                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");
                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");