if (!(vcpu->arch.aux_inuse & KVM_MIPS_AUX_FPU)) {
                __kvm_restore_fpu(&vcpu->arch);
                vcpu->arch.aux_inuse |= KVM_MIPS_AUX_FPU;
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_RESTORE, KVM_TRACE_AUX_FPU);
+       } else {
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_ENABLE, KVM_TRACE_AUX_FPU);
        }
 
        preempt_enable();
                 */
                __kvm_restore_msa_upper(&vcpu->arch);
                vcpu->arch.aux_inuse |= KVM_MIPS_AUX_MSA;
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_RESTORE, KVM_TRACE_AUX_MSA);
                break;
        case 0:
                /* Neither FPU or MSA already active, restore full MSA state */
                vcpu->arch.aux_inuse |= KVM_MIPS_AUX_MSA;
                if (kvm_mips_guest_has_fpu(&vcpu->arch))
                        vcpu->arch.aux_inuse |= KVM_MIPS_AUX_FPU;
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_RESTORE,
+                             KVM_TRACE_AUX_FPU_MSA);
                break;
        default:
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_ENABLE, KVM_TRACE_AUX_MSA);
                break;
        }
 
        preempt_disable();
        if (cpu_has_msa && vcpu->arch.aux_inuse & KVM_MIPS_AUX_MSA) {
                disable_msa();
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_DISCARD, KVM_TRACE_AUX_MSA);
                vcpu->arch.aux_inuse &= ~KVM_MIPS_AUX_MSA;
        }
        if (vcpu->arch.aux_inuse & KVM_MIPS_AUX_FPU) {
                clear_c0_status(ST0_CU1 | ST0_FR);
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_DISCARD, KVM_TRACE_AUX_FPU);
                vcpu->arch.aux_inuse &= ~KVM_MIPS_AUX_FPU;
        }
        preempt_enable();
                enable_fpu_hazard();
 
                __kvm_save_msa(&vcpu->arch);
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_SAVE, KVM_TRACE_AUX_FPU_MSA);
 
                /* Disable MSA & FPU */
                disable_msa();
 
                __kvm_save_fpu(&vcpu->arch);
                vcpu->arch.aux_inuse &= ~KVM_MIPS_AUX_FPU;
+               trace_kvm_aux(vcpu, KVM_TRACE_AUX_SAVE, KVM_TRACE_AUX_FPU);
 
                /* Disable FPU */
                clear_c0_status(ST0_CU1 | ST0_FR);
 
                      __entry->pc)
 );
 
+#define KVM_TRACE_AUX_RESTORE          0
+#define KVM_TRACE_AUX_SAVE             1
+#define KVM_TRACE_AUX_ENABLE           2
+#define KVM_TRACE_AUX_DISABLE          3
+#define KVM_TRACE_AUX_DISCARD          4
+
+#define KVM_TRACE_AUX_FPU              1
+#define KVM_TRACE_AUX_MSA              2
+#define KVM_TRACE_AUX_FPU_MSA          3
+
+#define kvm_trace_symbol_aux_op                \
+       { KVM_TRACE_AUX_RESTORE, "restore" },   \
+       { KVM_TRACE_AUX_SAVE,    "save" },      \
+       { KVM_TRACE_AUX_ENABLE,  "enable" },    \
+       { KVM_TRACE_AUX_DISABLE, "disable" },   \
+       { KVM_TRACE_AUX_DISCARD, "discard" }
+
+#define kvm_trace_symbol_aux_state             \
+       { KVM_TRACE_AUX_FPU,     "FPU" },       \
+       { KVM_TRACE_AUX_MSA,     "MSA" },       \
+       { KVM_TRACE_AUX_FPU_MSA, "FPU & MSA" }
+
+TRACE_EVENT(kvm_aux,
+           TP_PROTO(struct kvm_vcpu *vcpu, unsigned int op,
+                    unsigned int state),
+           TP_ARGS(vcpu, op, state),
+           TP_STRUCT__entry(
+                       __field(unsigned long, pc)
+                       __field(u8, op)
+                       __field(u8, state)
+           ),
+
+           TP_fast_assign(
+                       __entry->pc = vcpu->arch.pc;
+                       __entry->op = op;
+                       __entry->state = state;
+           ),
+
+           TP_printk("%s %s PC: 0x%08lx",
+                     __print_symbolic(__entry->op,
+                                      kvm_trace_symbol_aux_op),
+                     __print_symbolic(__entry->state,
+                                      kvm_trace_symbol_aux_state),
+                     __entry->pc)
+);
+
 #endif /* _TRACE_KVM_H */
 
 /* This part must be outside protection */