#define KRAIT_EVENT_MASK       (KRAIT_EVENT | VENUM_EVENT)
 #define PMRESRn_EN             BIT(31)
 
+#define EVENT_REGION(event)    (((event) >> 12) & 0xf)         /* R */
+#define EVENT_GROUP(event)     ((event) & 0xf)                 /* G */
+#define EVENT_CODE(event)      (((event) >> 4) & 0xff)         /* CC */
+#define EVENT_VENUM(event)     (!!(event & VENUM_EVENT))       /* N=2 */
+#define EVENT_CPU(event)       (!!(event & KRAIT_EVENT))       /* N=1 */
+
 static u32 krait_read_pmresrn(int n)
 {
        u32 val;
        }
 }
 
-static u32 krait_read_vpmresr0(void)
+static u32 venum_read_pmresr(void)
 {
        u32 val;
        asm volatile("mrc p10, 7, %0, c11, c0, 0" : "=r" (val));
        return val;
 }
 
-static void krait_write_vpmresr0(u32 val)
+static void venum_write_pmresr(u32 val)
 {
        asm volatile("mcr p10, 7, %0, c11, c0, 0" : : "r" (val));
 }
 
-static void krait_pre_vpmresr0(u32 *venum_orig_val, u32 *fp_orig_val)
+static void venum_pre_pmresr(u32 *venum_orig_val, u32 *fp_orig_val)
 {
        u32 venum_new_val;
        u32 fp_new_val;
        fmxr(FPEXC, fp_new_val);
 }
 
-static void krait_post_vpmresr0(u32 venum_orig_val, u32 fp_orig_val)
+static void venum_post_pmresr(u32 venum_orig_val, u32 fp_orig_val)
 {
        BUG_ON(preemptible());
        /* Restore FPEXC */
        u32 val;
        u32 mask;
        u32 vval, fval;
-       unsigned int region;
-       unsigned int group;
-       unsigned int code;
+       unsigned int region = EVENT_REGION(config_base);
+       unsigned int group = EVENT_GROUP(config_base);
+       unsigned int code = EVENT_CODE(config_base);
        unsigned int group_shift;
-       bool venum_event;
-
-       venum_event = !!(config_base & VENUM_EVENT);
-       region = (config_base >> 12) & 0xf;
-       code   = (config_base >> 4) & 0xff;
-       group  = (config_base >> 0)  & 0xf;
+       bool venum_event = EVENT_VENUM(config_base);
 
        group_shift = group * 8;
        mask = 0xff << group_shift;
        asm volatile("mcr p15, 0, %0, c9, c15, 0" : : "r" (0));
 
        if (venum_event) {
-               krait_pre_vpmresr0(&vval, &fval);
-               val = krait_read_vpmresr0();
+               venum_pre_pmresr(&vval, &fval);
+               val = venum_read_pmresr();
                val &= ~mask;
                val |= code << group_shift;
                val |= PMRESRn_EN;
-               krait_write_vpmresr0(val);
-               krait_post_vpmresr0(vval, fval);
+               venum_write_pmresr(val);
+               venum_post_pmresr(vval, fval);
        } else {
                val = krait_read_pmresrn(region);
                val &= ~mask;
        }
 }
 
-static u32 krait_clear_pmresrn_group(u32 val, int group)
+static u32 clear_pmresrn_group(u32 val, int group)
 {
        u32 mask;
        int group_shift;
 {
        u32 val;
        u32 vval, fval;
-       unsigned int region;
-       unsigned int group;
-       bool venum_event;
-
-       venum_event = !!(config_base & VENUM_EVENT);
-       region = (config_base >> 12) & 0xf;
-       group  = (config_base >> 0)  & 0xf;
+       unsigned int region = EVENT_REGION(config_base);
+       unsigned int group = EVENT_GROUP(config_base);
+       bool venum_event = EVENT_VENUM(config_base);
 
        if (venum_event) {
-               krait_pre_vpmresr0(&vval, &fval);
-               val = krait_read_vpmresr0();
-               val = krait_clear_pmresrn_group(val, group);
-               krait_write_vpmresr0(val);
-               krait_post_vpmresr0(vval, fval);
+               venum_pre_pmresr(&vval, &fval);
+               val = venum_read_pmresr();
+               val = clear_pmresrn_group(val, group);
+               venum_write_pmresr(val);
+               venum_post_pmresr(vval, fval);
        } else {
                val = krait_read_pmresrn(region);
-               val = krait_clear_pmresrn_group(val, group);
+               val = clear_pmresrn_group(val, group);
                krait_write_pmresrn(region, val);
        }
 }
        krait_write_pmresrn(1, 0);
        krait_write_pmresrn(2, 0);
 
-       krait_pre_vpmresr0(&vval, &fval);
-       krait_write_vpmresr0(0);
-       krait_post_vpmresr0(vval, fval);
+       venum_pre_pmresr(&vval, &fval);
+       venum_write_pmresr(0);
+       venum_post_pmresr(vval, fval);
 }
 
 static int krait_event_to_bit(struct perf_event *event, unsigned int region,
 {
        int idx;
        int bit = -1;
-       unsigned int prefix;
-       unsigned int region;
-       unsigned int code;
-       unsigned int group;
-       bool krait_event;
        struct hw_perf_event *hwc = &event->hw;
+       unsigned int region = EVENT_REGION(hwc->config_base);
+       unsigned int code = EVENT_CODE(hwc->config_base);
+       unsigned int group = EVENT_GROUP(hwc->config_base);
+       bool venum_event = EVENT_VENUM(hwc->config_base);
+       bool krait_event = EVENT_CPU(hwc->config_base);
 
-       region = (hwc->config_base >> 12) & 0xf;
-       code   = (hwc->config_base >> 4) & 0xff;
-       group  = (hwc->config_base >> 0) & 0xf;
-       krait_event = !!(hwc->config_base & KRAIT_EVENT_MASK);
-
-       if (krait_event) {
+       if (venum_event || krait_event) {
                /* Ignore invalid events */
                if (group > 3 || region > 2)
                        return -EINVAL;
-               prefix = hwc->config_base & KRAIT_EVENT_MASK;
-               if (prefix != KRAIT_EVENT && prefix != VENUM_EVENT)
-                       return -EINVAL;
-               if (prefix == VENUM_EVENT && (code & 0xe0))
+               if (venum_event && (code & 0xe0))
                        return -EINVAL;
 
                bit = krait_event_to_bit(event, region, group);
 {
        int bit;
        struct hw_perf_event *hwc = &event->hw;
-       unsigned int region;
-       unsigned int group;
-       bool krait_event;
-
-       region = (hwc->config_base >> 12) & 0xf;
-       group  = (hwc->config_base >> 0) & 0xf;
-       krait_event = !!(hwc->config_base & KRAIT_EVENT_MASK);
+       unsigned int region = EVENT_REGION(hwc->config_base);
+       unsigned int group = EVENT_GROUP(hwc->config_base);
+       bool venum_event = EVENT_VENUM(hwc->config_base);
+       bool krait_event = EVENT_CPU(hwc->config_base);
 
-       if (krait_event) {
+       if (venum_event || krait_event) {
                bit = krait_event_to_bit(event, region, group);
                clear_bit(bit, cpuc->used_mask);
        }