}
 
 static struct kvm_vm *create_vm(enum vm_guest_mode mode, uint32_t vcpuid,
-                               uint64_t extra_mem_pages, void *guest_code)
+                               uint64_t extra_mem_pages, void *guest_code,
+                               unsigned long type)
 {
        struct kvm_vm *vm;
        uint64_t extra_pg_pages = extra_mem_pages / 512 * 2;
 
-       vm = vm_create(mode, DEFAULT_GUEST_PHY_PAGES + extra_pg_pages, O_RDWR);
+       vm = _vm_create(mode, DEFAULT_GUEST_PHY_PAGES + extra_pg_pages,
+                       O_RDWR, type);
        kvm_vm_elf_load(vm, program_invocation_name, 0, 0);
 #ifdef __x86_64__
        vm_create_irqchip(vm);
        struct kvm_vm *vm;
        uint64_t max_gfn;
        unsigned long *bmap;
+       unsigned long type = 0;
 
        switch (mode) {
        case VM_MODE_P52V48_4K:
                guest_pa_bits = 52;
                guest_page_shift = 16;
                break;
+       case VM_MODE_P48V48_4K:
+               guest_pa_bits = 48;
+               guest_page_shift = 12;
+               break;
+       case VM_MODE_P48V48_64K:
+               guest_pa_bits = 48;
+               guest_page_shift = 16;
+               break;
        case VM_MODE_P40V48_4K:
                guest_pa_bits = 40;
                guest_page_shift = 12;
         * bits we can change to 39.
         */
        guest_pa_bits = 39;
+#endif
+#ifdef __aarch64__
+       if (guest_pa_bits != 40)
+               type = KVM_VM_TYPE_ARM_IPA_SIZE(guest_pa_bits);
 #endif
        max_gfn = (1ul << (guest_pa_bits - guest_page_shift)) - 1;
        guest_page_size = (1ul << guest_page_shift);
        bmap = bitmap_alloc(host_num_pages);
        host_bmap_track = bitmap_alloc(host_num_pages);
 
-       vm = create_vm(mode, VCPU_ID, guest_num_pages, guest_code);
+       vm = create_vm(mode, VCPU_ID, guest_num_pages, guest_code, type);
 
 #ifdef USE_CLEAR_DIRTY_LOG
        struct kvm_enable_cap cap = {};
        unsigned long interval = TEST_HOST_LOOP_INTERVAL;
        bool mode_selected = false;
        uint64_t phys_offset = 0;
-       unsigned int mode;
+       unsigned int mode, host_ipa_limit;
        int opt, i;
 
 #ifdef USE_CLEAR_DIRTY_LOG
 #ifdef __aarch64__
        vm_guest_mode_params_init(VM_MODE_P40V48_4K, true, true);
        vm_guest_mode_params_init(VM_MODE_P40V48_64K, true, true);
+
+       host_ipa_limit = kvm_check_cap(KVM_CAP_ARM_VM_IPA_SIZE);
+       if (host_ipa_limit >= 52)
+               vm_guest_mode_params_init(VM_MODE_P52V48_64K, true, true);
+       if (host_ipa_limit >= 48) {
+               vm_guest_mode_params_init(VM_MODE_P48V48_4K, true, true);
+               vm_guest_mode_params_init(VM_MODE_P48V48_64K, true, true);
+       }
 #endif
 
        while ((opt = getopt(argc, argv, "hi:I:p:m:")) != -1) {
 
 int vm_enable_cap(struct kvm_vm *vm, struct kvm_enable_cap *cap);
 
 struct kvm_vm *vm_create(enum vm_guest_mode mode, uint64_t phy_pages, int perm);
+struct kvm_vm *_vm_create(enum vm_guest_mode mode, uint64_t phy_pages,
+                         int perm, unsigned long type);
 void kvm_vm_free(struct kvm_vm *vmp);
 void kvm_vm_restart(struct kvm_vm *vmp, int perm);
 void kvm_vm_release(struct kvm_vm *vmp);
 
        return ret;
 }
 
-static void vm_open(struct kvm_vm *vm, int perm)
+static void vm_open(struct kvm_vm *vm, int perm, unsigned long type)
 {
        vm->kvm_fd = open(KVM_DEV_PATH, perm);
        if (vm->kvm_fd < 0)
                exit(KSFT_SKIP);
 
-       vm->fd = ioctl(vm->kvm_fd, KVM_CREATE_VM, NULL);
+       vm->fd = ioctl(vm->kvm_fd, KVM_CREATE_VM, type);
        TEST_ASSERT(vm->fd >= 0, "KVM_CREATE_VM ioctl failed, "
                "rc: %i errno: %i", vm->fd, errno);
 }
  * descriptor to control the created VM is created with the permissions
  * given by perm (e.g. O_RDWR).
  */
-struct kvm_vm *vm_create(enum vm_guest_mode mode, uint64_t phy_pages, int perm)
+struct kvm_vm *_vm_create(enum vm_guest_mode mode, uint64_t phy_pages,
+                         int perm, unsigned long type)
 {
        struct kvm_vm *vm;
        int kvm_fd;
        TEST_ASSERT(vm != NULL, "Insufficient Memory");
 
        vm->mode = mode;
-       vm_open(vm, perm);
+       vm->type = type;
+       vm_open(vm, perm, type);
 
        /* Setup mode specific traits. */
        switch (vm->mode) {
        return vm;
 }
 
+struct kvm_vm *vm_create(enum vm_guest_mode mode, uint64_t phy_pages, int perm)
+{
+       return _vm_create(mode, phy_pages, perm, 0);
+}
+
 /*
  * VM Restart
  *
 {
        struct userspace_mem_region *region;
 
-       vm_open(vmp, perm);
+       vm_open(vmp, perm, vmp->type);
        if (vmp->has_irqchip)
                vm_create_irqchip(vmp);