* Copyright © 2019 Intel Corporation
  */
 
+#include <drm/drm_managed.h>
+
 #include "intel_gt_debugfs.h"
 
 #include "gem/i915_gem_lmem.h"
        return 0;
 }
 
-void intel_gt_init_hw_early(struct intel_gt *gt, struct i915_ggtt *ggtt)
+int intel_gt_assign_ggtt(struct intel_gt *gt)
 {
-       gt->ggtt = ggtt;
+       gt->ggtt = drmm_kzalloc(>->i915->drm, sizeof(*gt->ggtt), GFP_KERNEL);
+
+       return gt->ggtt ? 0 : -ENOMEM;
 }
 
 static const struct intel_mmio_range icl_l3bank_steering_table[] = {
 
 
 void intel_gt_init_early(struct intel_gt *gt, struct drm_i915_private *i915);
 void __intel_gt_init_early(struct intel_gt *gt, struct drm_i915_private *i915);
-void intel_gt_init_hw_early(struct intel_gt *gt, struct i915_ggtt *ggtt);
+int intel_gt_assign_ggtt(struct intel_gt *gt);
 int intel_gt_probe_lmem(struct intel_gt *gt);
 int intel_gt_init_mmio(struct intel_gt *gt);
 int __must_check intel_gt_init_hw(struct intel_gt *gt);
 
 
        i915_perf_init(dev_priv);
 
-       intel_gt_init_hw_early(to_gt(dev_priv), &dev_priv->ggtt);
+       ret = intel_gt_assign_ggtt(to_gt(dev_priv));
+       if (ret)
+               goto err_perf;
 
        ret = i915_ggtt_probe_hw(dev_priv);
        if (ret)
 
        struct drm_atomic_state *modeset_restore_state;
        struct drm_modeset_acquire_ctx reset_ctx;
 
-       struct i915_ggtt ggtt; /* VM representing the global address space */
-
        struct i915_gem_mm mm;
 
        /* Kernel Modesetting */
 
                SUBTEST(igt_gtt_insert),
        };
        struct drm_i915_private *i915;
-       struct i915_ggtt *ggtt;
+       struct intel_gt *gt;
        int err;
 
        i915 = mock_gem_device();
        if (!i915)
                return -ENOMEM;
 
-       ggtt = kmalloc(sizeof(*ggtt), GFP_KERNEL);
-       if (!ggtt) {
-               err = -ENOMEM;
+       /* allocate the ggtt */
+       err = intel_gt_assign_ggtt(to_gt(i915));
+       if (err)
                goto out_put;
-       }
-       mock_init_ggtt(i915, ggtt);
 
-       err = i915_subtests(tests, ggtt);
+       gt = to_gt(i915);
+
+       mock_init_ggtt(gt);
+
+       err = i915_subtests(tests, gt->ggtt);
 
        mock_device_flush(i915);
        i915_gem_drain_freed_objects(i915);
-       mock_fini_ggtt(ggtt);
-       kfree(ggtt);
+       mock_fini_ggtt(gt->ggtt);
+
 out_put:
        mock_destroy_device(i915);
        return err;
 
                SUBTEST(igt_vma_partial),
        };
        struct drm_i915_private *i915;
-       struct i915_ggtt *ggtt;
+       struct intel_gt *gt;
        int err;
 
        i915 = mock_gem_device();
        if (!i915)
                return -ENOMEM;
 
-       ggtt = kmalloc(sizeof(*ggtt), GFP_KERNEL);
-       if (!ggtt) {
-               err = -ENOMEM;
+       /* allocate the ggtt */
+       err = intel_gt_assign_ggtt(to_gt(i915));
+       if (err)
                goto out_put;
-       }
-       mock_init_ggtt(i915, ggtt);
 
-       err = i915_subtests(tests, ggtt);
+       gt = to_gt(i915);
+
+       mock_init_ggtt(gt);
+
+       err = i915_subtests(tests, gt->ggtt);
 
        mock_device_flush(i915);
        i915_gem_drain_freed_objects(i915);
-       mock_fini_ggtt(ggtt);
-       kfree(ggtt);
+       mock_fini_ggtt(gt->ggtt);
+
 out_put:
        mock_destroy_device(i915);
        return err;
 
 
        mock_init_contexts(i915);
 
-       mock_init_ggtt(i915, &i915->ggtt);
-       to_gt(i915)->vm = i915_vm_get(&i915->ggtt.vm);
+       /* allocate the ggtt */
+       ret = intel_gt_assign_ggtt(to_gt(i915));
+       if (ret)
+               goto err_unlock;
+
+       mock_init_ggtt(to_gt(i915));
+       to_gt(i915)->vm = i915_vm_get(&to_gt(i915)->ggtt->vm);
 
        mkwrite_device_info(i915)->platform_engine_mask = BIT(0);
        to_gt(i915)->info.engine_mask = BIT(0);
 
 {
 }
 
-void mock_init_ggtt(struct drm_i915_private *i915, struct i915_ggtt *ggtt)
+void mock_init_ggtt(struct intel_gt *gt)
 {
-       memset(ggtt, 0, sizeof(*ggtt));
+       struct i915_ggtt *ggtt = gt->ggtt;
 
-       ggtt->vm.gt = to_gt(i915);
-       ggtt->vm.i915 = i915;
+       ggtt->vm.gt = gt;
+       ggtt->vm.i915 = gt->i915;
        ggtt->vm.is_ggtt = true;
 
        ggtt->gmadr = (struct resource) DEFINE_RES_MEM(0, 2048 * PAGE_SIZE);
        ggtt->vm.vma_ops.unbind_vma  = mock_unbind_ggtt;
 
        i915_address_space_init(&ggtt->vm, VM_CLASS_GGTT);
-       to_gt(i915)->ggtt = ggtt;
 }
 
 void mock_fini_ggtt(struct i915_ggtt *ggtt)
 
 
 struct drm_i915_private;
 struct i915_ggtt;
+struct intel_gt;
 
-void mock_init_ggtt(struct drm_i915_private *i915, struct i915_ggtt *ggtt);
+void mock_init_ggtt(struct intel_gt *gt);
 void mock_fini_ggtt(struct i915_ggtt *ggtt);
 
 struct i915_ppgtt *mock_ppgtt(struct drm_i915_private *i915, const char *name);