diff --git a/drivers/misc/habanalabs/Kconfig b/drivers/misc/habanalabs/Kconfig
index 80400a035dc1..a7cbc1fb6957 100644
--- a/drivers/misc/habanalabs/Kconfig
+++ b/drivers/misc/habanalabs/Kconfig
@@ -7,6 +7,7 @@ config HABANA_AI
 	depends on PCI
 	select FRAME_VECTOR
 	select DMA_SHARED_BUFFER
+	select GENERIC_ALLOCATOR
 	help
 	  Enables PCIe card driver for Habana's AI Processors (AIP) that are
 	  designed to accelerate Deep Learning inference and training workloads.