diff --git a/drivers/soc/qcom/mem_buf_dma_buf.c b/drivers/soc/qcom/mem_buf_dma_buf.c
index 1927a7bc70c5..1b15e35dae38 100644
--- a/drivers/soc/qcom/mem_buf_dma_buf.c
+++ b/drivers/soc/qcom/mem_buf_dma_buf.c
@@ -360,9 +360,11 @@ static int mem_buf_sgl_sync_range(struct device *dev, struct scatterlist *sgl,
 			break;
 
 		if (i > 0) {
-			pr_warn_ratelimited("Partial cmo only supported with 1 segment\n"
-				"is dma_set_max_seg_size being set on dev:%s\n",
-				dev_name(dev));
+			if (get_dma_ops(dev)) {
+				pr_warn_ratelimited("Partial cmo only supported with 1 segment\n"
+					"is dma_set_max_seg_size being set on dev:%s\n",
+					dev_name(dev));
+			}
 			return -EINVAL;
 		}
 	}
diff --git a/drivers/staging/android/ion/heaps/msm_ion_dma_buf.c b/drivers/staging/android/ion/heaps/msm_ion_dma_buf.c
index aba5c4d401db..6bd9dd286057 100644
--- a/drivers/staging/android/ion/heaps/msm_ion_dma_buf.c
+++ b/drivers/staging/android/ion/heaps/msm_ion_dma_buf.c
@@ -500,9 +500,11 @@ static int ion_sgl_sync_range(struct device *dev, struct scatterlist *sgl,
 			break;
 
 		if (i > 0) {
-			pr_warn_ratelimited("Partial cmo only supported with 1 segment\n"
-				"is dma_set_max_seg_size being set on dev:%s\n",
-				dev_name(dev));
+			if (get_dma_ops(dev)) {
+				pr_warn_ratelimited("Partial cmo only supported with 1 segment\n"
+					"is dma_set_max_seg_size being set on dev:%s\n",
+					dev_name(dev));
+			}
 			return -EINVAL;
 		}
 	}