diff --git a/BUILD.gn b/BUILD.gn
index 3d47cdb..bacd7e7 100644
--- a/BUILD.gn
+++ b/BUILD.gn
@@ -60,7 +60,7 @@
   origin_address = "0x06000000"
   boot_flow = "//src/boot_flow:spmc"
   console = "//src/arch/aarch64/pl011"
-  iommu = "//src/iommu:absent"
+  iommu = "//src/arch/aarch64/arm_smmuv3"
   gic_version = 3
   gicd_base_address = "0x2f000000"
   gicr_base_address = "0x2f100000"
@@ -73,6 +73,8 @@
     secure_world = "1"
     pl011_base_address = "0x1c090000"
     hftest_ctrl = "//test/hftest:ctrl_uart"
+    smmu_base_address = "0x2b400000"
+    smmu_memory_size = "0x100000"
   }
 }
 
@@ -81,7 +83,7 @@
   origin_address = "0x80000000"
   boot_flow = "//src/boot_flow:linux"
   console = "//src/arch/aarch64/pl011"
-  iommu = "//src/iommu:absent"
+  iommu = "//src/arch/aarch64/arm_smmuv3"
   gic_version = 3
   gicd_base_address = "0x2f000000"
   gicr_base_address = "0x2f100000"
@@ -90,6 +92,8 @@
   max_vms = 16
   toolchain_args = {
     pl011_base_address = "0x1c090000"
+    smmu_base_address = "0x2b400000"
+    smmu_memory_size = "0x100000"
   }
 }