session->header.env.cpu[i].socket_id);
        }
 
+       // Test that CPU ID contains socket, die, core and CPU
+       for (i = 0; i < map->nr; i++) {
+               id = aggr_cpu_id__cpu(perf_cpu_map__cpu(map, i), NULL);
+               TEST_ASSERT_VAL("Cpu map - CPU ID doesn't match", map->map[i] == id.cpu);
+
+               TEST_ASSERT_VAL("Cpu map - Core ID doesn't match",
+                       session->header.env.cpu[map->map[i]].core_id == id.core);
+               TEST_ASSERT_VAL("Cpu map - Socket ID doesn't match",
+                       session->header.env.cpu[map->map[i]].socket_id == id.socket);
+
+               TEST_ASSERT_VAL("Cpu map - Die ID doesn't match",
+                       session->header.env.cpu[map->map[i]].die_id == id.die);
+               TEST_ASSERT_VAL("Cpu map - Node ID is set", id.node == -1);
+               TEST_ASSERT_VAL("Cpu map - Thread is set", id.thread == -1);
+       }
+
        // Test that core ID contains socket, die and core
        for (i = 0; i < map->nr; i++) {
                id = aggr_cpu_id__core(perf_cpu_map__cpu(map, i), NULL);
 
                TEST_ASSERT_VAL("Die map - Node ID is set", id.node == -1);
                TEST_ASSERT_VAL("Die map - Core is set", id.core == -1);
+               TEST_ASSERT_VAL("Die map - CPU is set", id.cpu == -1);
                TEST_ASSERT_VAL("Die map - Thread is set", id.thread == -1);
        }
 
                TEST_ASSERT_VAL("Socket map - Node ID is set", id.node == -1);
                TEST_ASSERT_VAL("Socket map - Die ID is set", id.die == -1);
                TEST_ASSERT_VAL("Socket map - Core is set", id.core == -1);
+               TEST_ASSERT_VAL("Socket map - CPU is set", id.cpu == -1);
                TEST_ASSERT_VAL("Socket map - Thread is set", id.thread == -1);
        }
 
                TEST_ASSERT_VAL("Node map - Socket is set", id.socket == -1);
                TEST_ASSERT_VAL("Node map - Die ID is set", id.die == -1);
                TEST_ASSERT_VAL("Node map - Core is set", id.core == -1);
+               TEST_ASSERT_VAL("Node map - CPU is set", id.cpu == -1);
                TEST_ASSERT_VAL("Node map - Thread is set", id.thread == -1);
        }
        perf_session__delete(session);
 
        struct aggr_cpu_id id;
        int core = cpu__get_core_id(cpu);
 
-       /* aggr_cpu_id__die returns a struct with socket and die set*/
+       /* aggr_cpu_id__die returns a struct with socket and die set. */
        id = aggr_cpu_id__die(cpu, data);
        if (aggr_cpu_id__is_empty(&id))
                return id;
 
 }
 
+struct aggr_cpu_id aggr_cpu_id__cpu(int cpu, void *data)
+{
+       struct aggr_cpu_id id;
+
+       /* aggr_cpu_id__core returns a struct with socket, die and core set. */
+       id = aggr_cpu_id__core(cpu, data);
+       if (aggr_cpu_id__is_empty(&id))
+               return id;
+
+       id.cpu = cpu;
+       return id;
+
+}
+
 struct aggr_cpu_id aggr_cpu_id__node(int cpu, void *data __maybe_unused)
 {
        struct aggr_cpu_id id = aggr_cpu_id__empty();
                a->node == b->node &&
                a->socket == b->socket &&
                a->die == b->die &&
-               a->core == b->core;
+               a->core == b->core &&
+               a->cpu == b->cpu;
 }
 
 bool aggr_cpu_id__is_empty(const struct aggr_cpu_id *a)
                a->node == -1 &&
                a->socket == -1 &&
                a->die == -1 &&
-               a->core == -1;
+               a->core == -1 &&
+               a->cpu == -1;
 }
 
 struct aggr_cpu_id aggr_cpu_id__empty(void)
                .node = -1,
                .socket = -1,
                .die = -1,
-               .core = -1
+               .core = -1,
+               .cpu = -1
        };
        return ret;
 }
 
        int die;
        /** The core id as read from /sys/devices/system/cpu/cpuX/topology/core_id. */
        int core;
+       /** CPU aggregation, note there is one CPU for each SMT thread. */
+       int cpu;
 };
 
 /** A collection of aggr_cpu_id values, the "built" version is sorted and uniqued. */
  * compatible with aggr_cpu_id_get_t.
  */
 struct aggr_cpu_id aggr_cpu_id__core(int cpu, void *data);
+/**
+ * aggr_cpu_id__core - Create an aggr_cpu_id with the cpu, core, die and socket
+ * populated with the cpu, core, die and socket for cpu. The function signature
+ * is compatible with aggr_cpu_id_get_t.
+ */
+struct aggr_cpu_id aggr_cpu_id__cpu(int cpu, void *data);
 /**
  * aggr_cpu_id__node - Create an aggr_cpu_id with the numa node populated for
  * cpu. The function signature is compatible with aggr_cpu_id_get_t.