diff --git a/drivers/ub/ubfi/irq.c b/drivers/ub/ubfi/irq.c
index d47e4ce67bd6f08676042723e091dfc0a4899d68..eb716af1308846242058d57d1d0003d039063af0 100644
--- a/drivers/ub/ubfi/irq.c
+++ b/drivers/ub/ubfi/irq.c
@@ -12,6 +12,7 @@
 int ub_update_msi_domain(struct device *dev,
 			 enum irq_domain_bus_token bus_token)
 {
+#ifdef CONFIG_GENERIC_MSI_IRQ
 	struct fwnode_handle *fwnode;
 	struct irq_domain *domain;
 
@@ -35,7 +36,7 @@ int ub_update_msi_domain(struct device *dev,
 
 	/* Update msi domain with new bus_token */
 	dev_set_msi_domain(dev, domain);
-
+#endif
 	return 0;
 }
 EXPORT_SYMBOL_GPL(ub_update_msi_domain);