experiment with pci

Author: JBlaschke

Project.toml

@@ -8,7 +8,8 @@ AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
 CEnum = "fa961155-64e5-5f13-b03f-caf6b980ea82"
 CpuId = "adafc99b-e345-5852-983c-f28acb93d879"
 Sockets = "6462fe0b-24de-5631-8697-dd941f90decc"
-ThreadPools = "b189fb0b-2eb5-4ed4-bc0c-d34c51242431"
 
 [compat]
+AbstractTrees = "0.4"
 CEnum = "0.4"
+CpuId = "0.3"

test/Project.toml

@@ -0,0 +1,15 @@
+name = "NIC Test"
+uuid = "6f74fd91-2978-43ad-8164-3af8c0ec0142"
+authors = ["Johannes Blaschke <[email protected]>"]
+version = "0.1.0"
+
+[deps]
+AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
+CEnum = "fa961155-64e5-5f13-b03f-caf6b980ea82"
+CpuId = "adafc99b-e345-5852-983c-f28acb93d879"
+Sockets = "6462fe0b-24de-5631-8697-dd941f90decc"
+
+[compat]
+AbstractTrees = "0.4"
+CEnum = "0.4"
+CpuId = "0.3"

test/get_pci.jl

@@ -0,0 +1,60 @@
+using Hwloc, AbstractTrees
+
+x = gettopology()
+t = HwlocTreeNode{UInt8}(x)
+
+print_tree(t; maxdepth=10)
+
+function distance_to_core!(node, target_index)
+    # shield re-entrance when iterating
+    node.tag = 1
+
+    if node.type == :Core
+        if nodevalue(node).logical_index == target_index
+            return true, 0
+        end
+    end
+
+    for child in node.children
+        if child.tag == 1
+            continue
+        end
+
+        found, dist = distance_to_core!(child, target_index)
+        if found
+            return true, dist + 1
+        end
+
+    end
+
+    if node.parent != nothing
+        found, dist = distance_to_core!(node.parent, target_index)
+        if found
+            return true, dist + 1
+        end
+    end
+
+    return false, typemax(Int)
+end
+
+function distance_to_core(root, node, target_index)
+    Hwloc.tag_subtree!(t, 0) 
+    found, dist = distance_to_core!(node, target_index)
+    Hwloc.tag_subtree!(t, 0) 
+    return found, dist
+end
+
+pci_devs = Hwloc.get_nodes(t, :PCI_Device)
+
+network_devs = filter(
+    x->Hwloc.hwloc_pci_class_string(nodevalue(x).attr.class_id) == "Ethernet",
+    pci_devs
+)
+
+println(length(network_devs))
+
+for net in collect(network_devs)
+    found, dist = distance_to_core(t, net, 20)
+    print("$(dist): ")
+    print_tree(net)
+end