Not-so-pretty fix for logical numbering output. Needs improvement

Signed-off-by: Guillaume Mercier <[email protected]>

Author: GuillaumeMercier

src/qvi-bbuff-rmi.h

@@ -459,7 +459,7 @@ qvi_bbuff_rmi_pack_item_impl(
     }
     // Non-null data.
     char *datas = nullptr;
-    int rc = qvi_hwloc_bitmap_asprintf(data, &datas);
+    int rc = qvi_hwloc_bitmap_asprintf(NULL, data, &datas);
     if (qvi_unlikely(rc != QV_SUCCESS)) return rc;
     // We are sending the string representation of the cpuset.
     rc = buff->append(datas, strlen(datas) + 1);

src/qvi-hwloc.cc

@@ -180,15 +180,16 @@ qvi_hwloc_obj_type_is_host_resource(
 
 int
 qvi_hwloc_bitmap_string(
+    hwloc_topology_t topo,
     hwloc_const_bitmap_t bitmap,
     qv_bind_string_format_t format,
     char **result
 ) {
     switch (format) {
         case QV_BIND_STRING_AS_BITMAP:
-            return qvi_hwloc_bitmap_asprintf(bitmap, result);
+            return qvi_hwloc_bitmap_asprintf(topo, bitmap, result);
         case QV_BIND_STRING_AS_LIST:
-            return qvi_hwloc_bitmap_list_asprintf(bitmap, result);
+            return qvi_hwloc_bitmap_list_asprintf(topo, bitmap, result);
         default:
             *result = nullptr;
             return QV_ERR_INVLD_ARG;
@@ -816,7 +817,7 @@ qvi_hwloc_emit_cpubind(
     if (rc != QV_SUCCESS) return rc;
 
     char *cpusets = nullptr;
-    rc = qvi_hwloc_bitmap_asprintf(cpuset, &cpusets);
+    rc = qvi_hwloc_bitmap_asprintf(hwl->topo, cpuset, &cpusets);
     if (rc != QV_SUCCESS) goto out;
 
     qvi_log_info(
@@ -831,6 +832,7 @@ qvi_hwloc_emit_cpubind(
 
 int
 qvi_hwloc_bitmap_asprintf(
+    hwloc_topology_t topo,
     hwloc_const_cpuset_t cpuset,
     char **result
 ) {
@@ -840,12 +842,33 @@ qvi_hwloc_bitmap_asprintf(
         qvi_log_error("hwloc_bitmap_asprintf() failed");
         return QV_ERR_OOR;
     }
+
+    if (topo) {
+        int num_pus = hwloc_get_nbobjs_inside_cpuset_by_type(topo, cpuset, HWLOC_OBJ_PU);
+        hwloc_obj_t obj_pu = NULL;
+        char str[128] = {0};
+        sprintf(str,"%s"," | cpubind (logical)=[");
+        for(int idx = 0; idx < num_pus ; idx++){
+            char str2[64] = {0};
+            obj_pu = hwloc_get_next_obj_inside_cpuset_by_type(topo, cpuset, HWLOC_OBJ_PU, obj_pu);
+            sprintf(str2,"%i%s",obj_pu->logical_index, idx == (num_pus-1) ? "]" :",");
+            strcat(str,str2);
+        }
+
+        char *newstr = (char *)malloc(sizeof(char)*(strlen(str)+strlen(iresult)+1));
+        strcpy(newstr,iresult);
+        strcat(newstr,str);
+        free(iresult);
+        iresult = newstr;
+    }
+
     *result = iresult;
     return QV_SUCCESS;
 }
 
 int
 qvi_hwloc_bitmap_list_asprintf(
+    hwloc_topology_t topo,
     hwloc_const_cpuset_t cpuset,
     char **result
 ) {
@@ -855,6 +878,26 @@ qvi_hwloc_bitmap_list_asprintf(
         qvi_log_error("hwloc_bitmap_list_asprintf() failed");
         return QV_ERR_OOR;
     }
+
+    if (topo) {
+        int num_pus = hwloc_get_nbobjs_inside_cpuset_by_type(topo, cpuset, HWLOC_OBJ_PU);
+        hwloc_obj_t obj_pu = NULL;
+        char str[128] = {0};
+        sprintf(str,"%s"," | cpubind (logical)=[");
+        for(int idx = 0; idx < num_pus ; idx++){
+            char str2[64] = {0};
+            obj_pu = hwloc_get_next_obj_inside_cpuset_by_type(topo, cpuset, HWLOC_OBJ_PU, obj_pu);
+            sprintf(str2,"%i%s",obj_pu->logical_index, idx == (num_pus-1) ? "]" :",");
+            strcat(str,str2);
+        }
+
+        char *newstr = (char *)malloc(sizeof(char)*(strlen(str)+strlen(iresult)+1));
+        strcpy(newstr,iresult);
+        strcat(newstr,str);
+        free(iresult);
+        iresult = newstr;
+    }
+
     *result = iresult;
     return QV_SUCCESS;
 }
@@ -869,7 +912,7 @@ qvi_hwloc_cpuset_debug(
 #endif
     assert(cpuset);
     char *cpusets = nullptr;
-    int rc = qvi_hwloc_bitmap_asprintf(cpuset, &cpusets);
+    int rc = qvi_hwloc_bitmap_asprintf(NULL, cpuset, &cpusets);
     if (rc != QV_SUCCESS) {
         qvi_abort();
     }
@@ -957,7 +1000,7 @@ qvi_hwloc_task_get_cpubind_as_string(
     int rc = qvi_hwloc_task_get_cpubind(hwl, task_id, &cpuset);
     if (rc != QV_SUCCESS) return rc;
 
-    rc = qvi_hwloc_bitmap_asprintf(cpuset, cpusets);
+    rc = qvi_hwloc_bitmap_asprintf(hwl->topo, cpuset, cpusets);
     qvi_hwloc_bitmap_delete(&cpuset);
     return rc;
 }
@@ -1185,7 +1228,7 @@ qvi_hwloc_devices_emit(
     }
     for (auto &dev : *devlist) {
         char *cpusets = nullptr;
-        int rc = qvi_hwloc_bitmap_asprintf(dev->affinity.cdata(), &cpusets);
+        int rc = qvi_hwloc_bitmap_asprintf(hwl->topo, dev->affinity.cdata(), &cpusets);
         if (rc != QV_SUCCESS) return rc;
 
         qvi_log_info("  Device Name: {}", dev->name);

src/qvi-hwloc.h

@@ -187,6 +187,7 @@ qvi_hwloc_emit_cpubind(
  */
 int
 qvi_hwloc_bitmap_asprintf(
+    hwloc_topology_t topo,
     hwloc_const_cpuset_t cpuset,
     char **result
 );
@@ -196,6 +197,7 @@ qvi_hwloc_bitmap_asprintf(
  */
 int
 qvi_hwloc_bitmap_list_asprintf(
+    hwloc_topology_t topo,
     hwloc_const_cpuset_t cpuset,
     char **result
 );
@@ -537,6 +539,7 @@ qvi_hwloc_get_devices_in_bitmap(
 
 int
 qvi_hwloc_bitmap_string(
+    hwloc_topology_t topo,
     hwloc_const_bitmap_t bitmap,
     qv_bind_string_format_t format,
     char **result

src/qvi-scope.cc

@@ -213,7 +213,9 @@ qv_scope::bind_string(
     int rc = m_group->task()->bind_top(&bitmap);
     if (qvi_unlikely(rc != QV_SUCCESS)) return rc;
 
-    rc = qvi_hwloc_bitmap_string(bitmap, format, result);
+    hwloc_topology_t topo = qvi_hwloc_get_topo_obj(m_group->hwloc());
+
+    rc = qvi_hwloc_bitmap_string(topo, bitmap, format, result);
     qvi_hwloc_bitmap_delete(&bitmap);
     return rc;
 }

tests/internal/test-hwloc.c

@@ -208,7 +208,7 @@ main(void)
         ctu_panic("%s (rc=%s)", ers, qv_strerr(rc));
     }
 
-    rc = qvi_hwloc_bitmap_asprintf(bitmap, &binds);
+    rc = qvi_hwloc_bitmap_asprintf(NULL, bitmap, &binds);
     if (rc != QV_SUCCESS) {
         ers = "qvi_hwloc_bitmap_asprintf() failed";
         ctu_panic("%s (rc=%s)", ers, qv_strerr(rc));

tests/internal/test-rmi.cc

@@ -112,7 +112,7 @@ client(
     }
 
     char *res;
-    qvi_hwloc_bitmap_asprintf(bitmap, &res);
+    qvi_hwloc_bitmap_asprintf(NULL, bitmap, &res);
     printf("# [%d] cpubind = %s\n", who, res);
     hwloc_bitmap_free(bitmap);
     free(res);