diff --git a/sun-ipa.dtsi b/sun-ipa.dtsi
index a19e824f..bba7776c 100644
--- a/sun-ipa.dtsi
+++ b/sun-ipa.dtsi
@@ -44,12 +44,19 @@
 
         /* Low Latency pipe alloc factor */
         qcom,ipa-gen-rx-ll-pool-sz-factor = <1>;
+	#address-cells = <2>;
+	#size-cells = <2>;
 };
 
 &soc {
 	ipa_smmu_ap_partition: ipa_smmu_ap_partition {
-		iommu-addresses = <&ipa_smmu_ap 0x0 0x20000000>,
-			<&ipa_smmu_ap 0x40000000 0xC0000000>;
+		iommu-addresses = <&ipa_smmu_ap 0x0 0x0 0x0 0x20000000>,
+			<&ipa_smmu_ap 0x0 0x40000000 0xFFFFFFFF 0xC0000000>;
+	};
+
+	ipa_smmu_uc_partition: ipa_smmu_uc_partition {
+		iommu-addresses = <&ipa_smmu_uc 0x0 0x0 0x0 0x20000000>,
+			<&ipa_smmu_uc 0x0 0x40000000 0xFFFFFFFF 0xC0000000>;
 	};
 };
 
@@ -60,3 +67,8 @@
 		/* modem tables in IMEM */
 		<0x14683000 0x14683000 0x2000>;
 };
+
+&ipa_smmu_uc {
+	/delete-property/ qcom,iommu-dma-addr-pool;
+	memory-region = <&ipa_smmu_uc_partition>;
+};