diff --git a/drivers/pci/ats.c b/drivers/pci/ats.c
index 7ec56fb0bd78aca5aec916aefd8788cc1b101027..b0dd08e6a9da1cc4f8ee78ebf53f1e4e5f1de2b2 100644
--- a/drivers/pci/ats.c
+++ b/drivers/pci/ats.c
@@ -13,6 +13,7 @@
 #include <linux/export.h>
 #include <linux/pci-ats.h>
 #include <linux/pci.h>
+#include <linux/slab.h>
 
 #include "pci.h"