1 files changed, 6 insertions, 1 deletions

diff --git a/arch/s390/kernel/topology.c b/arch/s390/kernel/topology.c
index 247b4c2d1e51..bcef00766a64 100644
--- a/arch/s390/kernel/topology.c
+++ b/arch/s390/kernel/topology.c
@@ -37,7 +37,8 @@ struct tl_cpu {
 };
 
 struct tl_container {
-        unsigned char reserved[8];
+        unsigned char reserved[7];
+        unsigned char id;
 };
 
 union tl_entry {
@@ -58,6 +59,7 @@ struct tl_info {
 
 struct core_info {
         struct core_info *next;
+        unsigned char id;
         cpumask_t mask;
 };
 
@@ -73,6 +75,7 @@ static DECLARE_WORK(topology_work, topology_work_fn);
 static DEFINE_SPINLOCK(topology_lock);
 
 cpumask_t cpu_core_map[NR_CPUS];
+unsigned char cpu_core_id[NR_CPUS];
 
 static cpumask_t cpu_coregroup_map(unsigned int cpu)
 {
@@ -116,6 +119,7 @@ static void add_cpus_to_core(struct tl_cpu *tl_cpu, struct core_info *core)
                 for_each_present_cpu(lcpu) {
                         if (cpu_logical_map(lcpu) == rcpu) {
                                 cpu_set(lcpu, core->mask);
+                                cpu_core_id[lcpu] = core->id;
                                 smp_cpu_polarization[lcpu] = tl_cpu->pp;
                         }
                 }
@@ -158,6 +162,7 @@ static void tl_to_cores(struct tl_info *info)
                         break;
                 case 1:
                         core = core->next;
+                        core->id = tle->container.id;
                         break;
                 case 0:
                         add_cpus_to_core(&tle->cpu, core);