1 files changed, 4 insertions, 3 deletions

diff --git a/arch/x86/kvm/lapic.c b/arch/x86/kvm/lapic.c
index 4b6c2da7265c..991fdf7fc17f 100644
--- a/arch/x86/kvm/lapic.c
+++ b/arch/x86/kvm/lapic.c
@@ -181,7 +181,8 @@ static void recalculate_apic_map(struct kvm *kvm)
                         max_id = max(max_id, kvm_x2apic_id(vcpu->arch.apic));
 
         new = kvzalloc(sizeof(struct kvm_apic_map) +
-                           sizeof(struct kvm_lapic *) * ((u64)max_id + 1), GFP_KERNEL);
+                           sizeof(struct kvm_lapic *) * ((u64)max_id + 1),
+                           GFP_KERNEL_ACCOUNT);
 
         if (!new)
                 goto out;
@@ -2259,13 +2260,13 @@ int kvm_create_lapic(struct kvm_vcpu *vcpu)
         ASSERT(vcpu != NULL);
         apic_debug("apic_init %d\n", vcpu->vcpu_id);
 
-        apic = kzalloc(sizeof(*apic), GFP_KERNEL);
+        apic = kzalloc(sizeof(*apic), GFP_KERNEL_ACCOUNT);
         if (!apic)
                 goto nomem;
 
         vcpu->arch.apic = apic;
 
-        apic->regs = (void *)get_zeroed_page(GFP_KERNEL);
+        apic->regs = (void *)get_zeroed_page(GFP_KERNEL_ACCOUNT);
         if (!apic->regs) {
                 printk(KERN_ERR "malloc apic regs error for vcpu %x\n",
                        vcpu->vcpu_id);