diff --git a/sun-ipa.dtsi b/sun-ipa.dtsi
index 2a25e744..a19e824f 100644
--- a/sun-ipa.dtsi
+++ b/sun-ipa.dtsi
@@ -46,8 +46,17 @@
         qcom,ipa-gen-rx-ll-pool-sz-factor = <1>;
 };
 
-&ipa_smmu_ap {
-        qcom,additional-mapping =
-                /* modem tables in IMEM */
-                <0x14683000 0x14683000 0x2000>;
+&soc {
+	ipa_smmu_ap_partition: ipa_smmu_ap_partition {
+		iommu-addresses = <&ipa_smmu_ap 0x0 0x20000000>,
+			<&ipa_smmu_ap 0x40000000 0xC0000000>;
+	};
+};
+
+&ipa_smmu_ap {
+	/delete-property/ qcom,iommu-dma-addr-pool;
+	memory-region = <&ipa_smmu_ap_partition>;
+	qcom,additional-mapping =
+		/* modem tables in IMEM */
+		<0x14683000 0x14683000 0x2000>;
 };