static inline void update_halt_poll_stats(struct kvm_vcpu *vcpu, ktime_t start,
                                          ktime_t end, bool success)
 {
+       struct kvm_vcpu_stat_generic *stats = &vcpu->stat.generic;
        u64 poll_ns = ktime_to_ns(ktime_sub(end, start));
 
-       if (success)
-               vcpu->stat.generic.halt_poll_success_ns += poll_ns;
-       else
-               vcpu->stat.generic.halt_poll_fail_ns += poll_ns;
+       ++vcpu->stat.generic.halt_attempted_poll;
+
+       if (success) {
+               ++vcpu->stat.generic.halt_successful_poll;
+
+               if (!vcpu_valid_wakeup(vcpu))
+                       ++vcpu->stat.generic.halt_poll_invalid;
+
+               stats->halt_poll_success_ns += poll_ns;
+               KVM_STATS_LOG_HIST_UPDATE(stats->halt_poll_success_hist, poll_ns);
+       } else {
+               stats->halt_poll_fail_ns += poll_ns;
+               KVM_STATS_LOG_HIST_UPDATE(stats->halt_poll_fail_hist, poll_ns);
+       }
 }
 
 /*
        if (do_halt_poll) {
                ktime_t stop = ktime_add_ns(ktime_get(), vcpu->halt_poll_ns);
 
-               ++vcpu->stat.generic.halt_attempted_poll;
                do {
                        /*
                         * This sets KVM_REQ_UNHALT if an interrupt
                         * arrives.
                         */
-                       if (kvm_vcpu_check_block(vcpu) < 0) {
-                               ++vcpu->stat.generic.halt_successful_poll;
-                               if (!vcpu_valid_wakeup(vcpu))
-                                       ++vcpu->stat.generic.halt_poll_invalid;
-
-                               KVM_STATS_LOG_HIST_UPDATE(
-                                     vcpu->stat.generic.halt_poll_success_hist,
-                                     ktime_to_ns(ktime_get()) -
-                                     ktime_to_ns(start));
+                       if (kvm_vcpu_check_block(vcpu) < 0)
                                goto out;
-                       }
                        cpu_relax();
                        poll_end = cur = ktime_get();
                } while (kvm_vcpu_can_poll(cur, stop));
-
-               KVM_STATS_LOG_HIST_UPDATE(
-                               vcpu->stat.generic.halt_poll_fail_hist,
-                               ktime_to_ns(ktime_get()) - ktime_to_ns(start));
        }