1 files changed, 7 insertions, 0 deletions

diff --git a/arch/powerpc/kvm/e500.c b/arch/powerpc/kvm/e500.c
index 64949eef43f1..669a5c5fc7d7 100644
--- a/arch/powerpc/kvm/e500.c
+++ b/arch/powerpc/kvm/e500.c
@@ -13,6 +13,7 @@
  */
 
 #include <linux/kvm_host.h>
+#include <linux/slab.h>
 #include <linux/err.h>
 
 #include <asm/reg.h>
@@ -60,6 +61,12 @@ int kvmppc_core_vcpu_setup(struct kvm_vcpu *vcpu)
 
         kvmppc_e500_tlb_setup(vcpu_e500);
 
+        /* Registers init */
+        vcpu->arch.pvr = mfspr(SPRN_PVR);
+
+        /* Since booke kvm only support one core, update all vcpus' PIR to 0 */
+        vcpu->vcpu_id = 0;
+
         return 0;
 }