diff --git a/drivers/infiniband/hw/hns/roce-customer/rdfx_entry.c b/drivers/infiniband/hw/hns/roce-customer/rdfx_entry.c
index 52cb9f7e088b4c4709eac7728d49475f3c4f38ef..8c2de43bcf14b2bb23bd832e5be458913d8f9201 100644
--- a/drivers/infiniband/hw/hns/roce-customer/rdfx_entry.c
+++ b/drivers/infiniband/hw/hns/roce-customer/rdfx_entry.c
@@ -1,6 +1,6 @@
 // SPDX-License-Identifier: GPL-2.0+
 // Copyright (c) 2016-2017 Hisilicon Limited.
-
+#ifdef CONFIG_INFINIBAND_HNS_DFX
 #include "roce_k_compat.h"
 #include "rdfx_common.h"
 #include "rdfx_intf.h"
@@ -328,3 +328,4 @@ void rdfx_set_cqe_info(struct hns_roce_dev *hr_dev, struct hns_roce_cq *hr_cq,
 	}
 }
 EXPORT_SYMBOL_GPL(rdfx_set_cqe_info);
+#endif
diff --git a/drivers/infiniband/hw/hns/roce-customer/rdfx_intf.c b/drivers/infiniband/hw/hns/roce-customer/rdfx_intf.c
index 7561e0ef5ebf3a1252df3cf68c3fceae2274995d..9a84e7ce6417022ad5ba0a45c7ebdedc4b9c7869 100644
--- a/drivers/infiniband/hw/hns/roce-customer/rdfx_intf.c
+++ b/drivers/infiniband/hw/hns/roce-customer/rdfx_intf.c
@@ -1,6 +1,6 @@
 // SPDX-License-Identifier: GPL-2.0+
 // Copyright (c) 2016-2017 Hisilicon Limited.
-
+#ifdef CONFIG_INFINIBAND_HNS_DFX
 #include "roce_k_compat.h"
 #include "hns_roce_device.h"
 #include "hns_roce_common.h"
@@ -940,4 +940,4 @@ void rdfx_set_rdfx_cq_ci(struct hns_roce_dev *hr_dev,
 	spin_unlock(&rdfx->cq.cq_lock);
 }
 EXPORT_SYMBOL_GPL(rdfx_set_rdfx_cq_ci);
-
+#endif