r = 0;
                break;
        case KVM_CAP_S390_VECTOR_REGISTERS:
-               kvm->arch.use_vectors = MACHINE_HAS_VX;
-               r = MACHINE_HAS_VX ? 0 : -EINVAL;
+               if (MACHINE_HAS_VX) {
+                       set_kvm_facility(kvm->arch.model.fac->mask, 129);
+                       set_kvm_facility(kvm->arch.model.fac->list, 129);
+                       r = 0;
+               } else
+                       r = -EINVAL;
                break;
        case KVM_CAP_S390_USER_STSI:
                kvm->arch.user_stsi = 1;
 
        kvm->arch.css_support = 0;
        kvm->arch.use_irqchip = 0;
-       kvm->arch.use_vectors = 0;
        kvm->arch.epoch = 0;
 
        spin_lock_init(&kvm->arch.start_stop_lock);
 void kvm_arch_vcpu_load(struct kvm_vcpu *vcpu, int cpu)
 {
        save_fp_ctl(&vcpu->arch.host_fpregs.fpc);
-       if (vcpu->kvm->arch.use_vectors)
+       if (test_kvm_facility(vcpu->kvm, 129))
                save_vx_regs((__vector128 *)&vcpu->arch.host_vregs->vrs);
        else
                save_fp_regs(vcpu->arch.host_fpregs.fprs);
        save_access_regs(vcpu->arch.host_acrs);
-       if (vcpu->kvm->arch.use_vectors) {
+       if (test_kvm_facility(vcpu->kvm, 129)) {
                restore_fp_ctl(&vcpu->run->s.regs.fpc);
                restore_vx_regs((__vector128 *)&vcpu->run->s.regs.vrs);
        } else {
 {
        atomic_clear_mask(CPUSTAT_RUNNING, &vcpu->arch.sie_block->cpuflags);
        gmap_disable(vcpu->arch.gmap);
-       if (vcpu->kvm->arch.use_vectors) {
+       if (test_kvm_facility(vcpu->kvm, 129)) {
                save_fp_ctl(&vcpu->run->s.regs.fpc);
                save_vx_regs((__vector128 *)&vcpu->run->s.regs.vrs);
        } else {
        }
        save_access_regs(vcpu->run->s.regs.acrs);
        restore_fp_ctl(&vcpu->arch.host_fpregs.fpc);
-       if (vcpu->kvm->arch.use_vectors)
+       if (test_kvm_facility(vcpu->kvm, 129))
                restore_vx_regs((__vector128 *)&vcpu->arch.host_vregs->vrs);
        else
                restore_fp_regs(vcpu->arch.host_fpregs.fprs);
                vcpu->arch.sie_block->eca |= 1;
        if (sclp_has_sigpif())
                vcpu->arch.sie_block->eca |= 0x10000000U;
-       if (vcpu->kvm->arch.use_vectors) {
+       if (test_kvm_facility(vcpu->kvm, 129)) {
                vcpu->arch.sie_block->eca |= 0x00020000;
                vcpu->arch.sie_block->ecd |= 0x20000000;
        }
 
                __test_facility(nr, kvm->arch.model.fac->list);
 }
 
+static inline int set_kvm_facility(u64 *fac_list, unsigned long nr)
+{
+       unsigned char *ptr;
+
+       if (nr >= MAX_FACILITY_BIT)
+               return -EINVAL;
+       ptr = (unsigned char *) fac_list + (nr >> 3);
+       *ptr |= (0x80UL >> (nr & 7));
+       return 0;
+}
+
 /* are cpu states controlled by user space */
 static inline int kvm_s390_user_cpu_state_ctrl(struct kvm *kvm)
 {