}
 }
 
-static void __set_cpuflag(struct kvm_vcpu *vcpu, u32 flag)
-{
-       atomic_or(flag, &vcpu->arch.sie_block->cpuflags);
-}
-
 static void set_intercept_indicators_io(struct kvm_vcpu *vcpu)
 {
        if (!(pending_irqs(vcpu) & IRQ_PEND_IO_MASK))
                return;
        else if (psw_ioint_disabled(vcpu))
-               __set_cpuflag(vcpu, CPUSTAT_IO_INT);
+               kvm_s390_set_cpuflags(vcpu, CPUSTAT_IO_INT);
        else
                vcpu->arch.sie_block->lctl |= LCTL_CR6;
 }
        if (!(pending_irqs(vcpu) & IRQ_PEND_EXT_MASK))
                return;
        if (psw_extint_disabled(vcpu))
-               __set_cpuflag(vcpu, CPUSTAT_EXT_INT);
+               kvm_s390_set_cpuflags(vcpu, CPUSTAT_EXT_INT);
        else
                vcpu->arch.sie_block->lctl |= LCTL_CR0;
 }
 static void set_intercept_indicators_stop(struct kvm_vcpu *vcpu)
 {
        if (kvm_s390_is_stop_irq_pending(vcpu))
-               __set_cpuflag(vcpu, CPUSTAT_STOP_INT);
+               kvm_s390_set_cpuflags(vcpu, CPUSTAT_STOP_INT);
 }
 
 /* Set interception request for non-deliverable interrupts */
 
        li->irq.ext = irq->u.ext;
        set_bit(IRQ_PEND_PFAULT_INIT, &li->pending_irqs);
-       __set_cpuflag(vcpu, CPUSTAT_EXT_INT);
+       kvm_s390_set_cpuflags(vcpu, CPUSTAT_EXT_INT);
        return 0;
 }
 
        if (test_and_set_bit(IRQ_PEND_EXT_EXTERNAL, &li->pending_irqs))
                return -EBUSY;
        *extcall = irq->u.extcall;
-       __set_cpuflag(vcpu, CPUSTAT_EXT_INT);
+       kvm_s390_set_cpuflags(vcpu, CPUSTAT_EXT_INT);
        return 0;
 }
 
        if (test_and_set_bit(IRQ_PEND_SIGP_STOP, &li->pending_irqs))
                return -EBUSY;
        stop->flags = irq->u.stop.flags;
-       __set_cpuflag(vcpu, CPUSTAT_STOP_INT);
+       kvm_s390_set_cpuflags(vcpu, CPUSTAT_STOP_INT);
        return 0;
 }
 
 
        set_bit(irq->u.emerg.code, li->sigp_emerg_pending);
        set_bit(IRQ_PEND_EXT_EMERGENCY, &li->pending_irqs);
-       __set_cpuflag(vcpu, CPUSTAT_EXT_INT);
+       kvm_s390_set_cpuflags(vcpu, CPUSTAT_EXT_INT);
        return 0;
 }
 
                                   0, 0);
 
        set_bit(IRQ_PEND_EXT_CLOCK_COMP, &li->pending_irqs);
-       __set_cpuflag(vcpu, CPUSTAT_EXT_INT);
+       kvm_s390_set_cpuflags(vcpu, CPUSTAT_EXT_INT);
        return 0;
 }
 
                                   0, 0);
 
        set_bit(IRQ_PEND_EXT_CPU_TIMER, &li->pending_irqs);
-       __set_cpuflag(vcpu, CPUSTAT_EXT_INT);
+       kvm_s390_set_cpuflags(vcpu, CPUSTAT_EXT_INT);
        return 0;
 }
 
        /* make the VCPU drop out of the SIE, or wake it up if sleeping */
        switch (type) {
        case KVM_S390_MCHK:
-               __set_cpuflag(dst_vcpu, CPUSTAT_STOP_INT);
+               kvm_s390_set_cpuflags(dst_vcpu, CPUSTAT_STOP_INT);
                break;
        case KVM_S390_INT_IO_MIN...KVM_S390_INT_IO_MAX:
-               __set_cpuflag(dst_vcpu, CPUSTAT_IO_INT);
+               kvm_s390_set_cpuflags(dst_vcpu, CPUSTAT_IO_INT);
                break;
        default:
-               __set_cpuflag(dst_vcpu, CPUSTAT_EXT_INT);
+               kvm_s390_set_cpuflags(dst_vcpu, CPUSTAT_EXT_INT);
                break;
        }
        kvm_s390_vcpu_wakeup(dst_vcpu);