KVM: Use kmem cache for allocating vcpus

Avi wants the allocations of vcpus centralized again.  The easiest way
is to add a "size" arg to kvm_init_arch, and expose the thus-prepared
cache to the modules.

Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
Signed-off-by: Avi Kivity <avi@qumranet.com>

1 files changed, 3 insertions, 2 deletions

diff --git a/drivers/kvm/svm.c b/drivers/kvm/svm.c
index b25f4e117e71..8193651dd815 100644
--- a/drivers/kvm/svm.c
+++ b/drivers/kvm/svm.c
@@ -577,7 +577,7 @@ static struct kvm_vcpu *svm_create_vcpu(struct kvm *kvm, unsigned int id)
         struct page *page;
         int err;
 
-        svm = kzalloc(sizeof *svm, GFP_KERNEL);
+        svm = kmem_cache_zalloc(kvm_vcpu_cache, GFP_KERNEL);
         if (!svm) {
                 err = -ENOMEM;
                 goto out;
@@ -1849,7 +1849,8 @@ static struct kvm_arch_ops svm_arch_ops = {
 
 static int __init svm_init(void)
 {
-        return kvm_init_arch(&svm_arch_ops, THIS_MODULE);
+        return kvm_init_arch(&svm_arch_ops, sizeof(struct vcpu_svm),
+                              THIS_MODULE);
 }
 
 static void __exit svm_exit(void)