diff --git a/qcom/diwali-dma-heaps.dtsi b/qcom/diwali-dma-heaps.dtsi
index 0ae9fffa..b7faba12 100644
--- a/qcom/diwali-dma-heaps.dtsi
+++ b/qcom/diwali-dma-heaps.dtsi
@@ -35,6 +35,20 @@
 			memory-region = <&qseecom_ta_mem>;
 		};
 
+		qcom,display {
+			qcom,dma-heap-name = "qcom,display";
+			qcom,dma-heap-type = <HEAP_TYPE_CMA>;
+			qcom,max-align = <9>;
+			memory-region = <&non_secure_display_memory>;
+		};
+
+		qcom,demura {
+			qcom,dma-heap-name = "qcom,demura";
+			qcom,dma-heap-type = <HEAP_TYPE_CMA>;
+			qcom,max-align = <9>;
+			memory-region = <&demura_heap_memory>;
+		};
+
 		qcom,audio_ml {
 			qcom,dma-heap-name = "qcom,audio-ml";
 			qcom,dma-heap-type = <HEAP_TYPE_CMA>;
diff --git a/qcom/diwali.dtsi b/qcom/diwali.dtsi
index 45976b9b..c49049c8 100644
--- a/qcom/diwali.dtsi
+++ b/qcom/diwali.dtsi
@@ -384,6 +384,22 @@
 		mem-type = <2>;
 	};
 
+	non_secure_display_memory: non_secure_display_region {
+		compatible = "shared-dma-pool";
+		reusable;
+		alloc-ranges = <0x0 0x00000000 0x0 0xffffffff>;
+		size = <0x0 0xa400000>;
+		alignment = <0x0 0x400000>;
+	};
+
+	demura_heap_memory: demura_heap_region {
+		compatible = "shared-dma-pool";
+		reusable;
+		alloc-ranges = <0x1 0x00000000 0xfffffffe 0xffffffff>;
+		size = <0x0 0x2800000>;
+		alignment = <0x0 0x400000>;
+	};
+
 	va_md_mem: va_md_mem_region {
 		compatible = "shared-dma-pool";
 		alloc-ranges = <0x1 0x00000000 0xfffffffe 0xffffffff>;