struct kvm *kvm_arch_alloc_vm(void)
  {
+       size_t sz = sizeof(struct kvm);
+ 
        if (!has_vhe())
-               return kzalloc(sizeof(struct kvm), GFP_KERNEL);
+               return kzalloc(sz, GFP_KERNEL_ACCOUNT);
  
-       return vzalloc(sizeof(struct kvm));
+       return __vmalloc(sz, GFP_KERNEL_ACCOUNT | __GFP_HIGHMEM | __GFP_ZERO);
  }
  
 -void kvm_arch_free_vm(struct kvm *kvm)
 -{
 -      if (!has_vhe())
 -              kfree(kvm);
 -      else
 -              vfree(kvm);
 -}
 -
  int kvm_arch_vcpu_precreate(struct kvm *kvm, unsigned int id)
  {
        if (irqchip_in_kernel(kvm) && vgic_initialized(kvm))
 
  TEST_GEN_PROGS_x86_64 += set_memory_region_test
  TEST_GEN_PROGS_x86_64 += steal_time
  TEST_GEN_PROGS_x86_64 += kvm_binary_stats_test
 +TEST_GEN_PROGS_x86_64 += system_counter_offset_test
  
+ TEST_GEN_PROGS_aarch64 += aarch64/arch_timer
  TEST_GEN_PROGS_aarch64 += aarch64/debug-exceptions
  TEST_GEN_PROGS_aarch64 += aarch64/get-reg-list
  TEST_GEN_PROGS_aarch64 += aarch64/psci_cpu_on_test