iommu/amd: Split enable_iommus() routine

Split the enable_iommus() routine so that a part of it can
run in early code.

Signed-off-by: Joerg Roedel <joerg.roedel@amd.com>

diff --git a/drivers/iommu/amd_iommu_init.c b/drivers/iommu/amd_iommu_init.c
index a5dbefb8e260e769f99c3c4790c7348839725a4b..b82cf447724e7005ec97182b166ba8712a526114 100644 (file)
--- a/drivers/iommu/amd_iommu_init.c
+++ b/drivers/iommu/amd_iommu_init.c
@@ -1379,7 +1379,7 @@ static void iommu_apply_resume_quirks(struct amd_iommu *iommu)
  * This function finally enables all IOMMUs found in the system after
  * they have been initialized
  */
-static void enable_iommus(void)
+static void early_enable_iommus(void)
 {
        struct amd_iommu *iommu;
 
@@ -1389,14 +1389,29 @@ static void enable_iommus(void)
                iommu_set_device_table(iommu);
                iommu_enable_command_buffer(iommu);
                iommu_enable_event_buffer(iommu);
-               iommu_enable_ppr_log(iommu);
-               iommu_enable_gt(iommu);
                iommu_set_exclusion_range(iommu);
                iommu_enable(iommu);
                iommu_flush_all_caches(iommu);
        }
 }
 
+static void enable_iommus_v2(void)
+{
+       struct amd_iommu *iommu;
+
+       for_each_iommu(iommu) {
+               iommu_enable_ppr_log(iommu);
+               iommu_enable_gt(iommu);
+       }
+}
+
+static void enable_iommus(void)
+{
+       early_enable_iommus();
+
+       enable_iommus_v2();
+}
+
 static void disable_iommus(void)
 {
        struct amd_iommu *iommu;