| .. | .. |
|---|
| 725 | 725 | * involves poking the GIC, which must be done in a |
|---|
| 726 | 726 | * non-preemptible context. |
|---|
| 727 | 727 | */ |
|---|
| 728 | | - preempt_disable(); |
|---|
| 728 | + migrate_disable(); |
|---|
| 729 | 729 | |
|---|
| 730 | 730 | kvm_pmu_flush_hwstate(vcpu); |
|---|
| 731 | 731 | |
|---|
| .. | .. |
|---|
| 774 | 774 | kvm_timer_sync_hwstate(vcpu); |
|---|
| 775 | 775 | kvm_vgic_sync_hwstate(vcpu); |
|---|
| 776 | 776 | local_irq_enable(); |
|---|
| 777 | | - preempt_enable(); |
|---|
| 777 | + migrate_enable(); |
|---|
| 778 | 778 | continue; |
|---|
| 779 | 779 | } |
|---|
| 780 | 780 | |
|---|
| .. | .. |
|---|
| 852 | 852 | /* Exit types that need handling before we can be preempted */ |
|---|
| 853 | 853 | handle_exit_early(vcpu, run, ret); |
|---|
| 854 | 854 | |
|---|
| 855 | | - preempt_enable(); |
|---|
| 855 | + migrate_enable(); |
|---|
| 856 | 856 | |
|---|
| 857 | 857 | ret = handle_exit(vcpu, run, ret); |
|---|
| 858 | 858 | } |
|---|
