refs #2: Fix debug config to use generalized GPU ordering

ANLAB-KAIST · May 12, 2015 · 566ea8a · 566ea8a
1 parent 588790d
commit 566ea8a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/configs/debug.py b/configs/debug.py
@@ -4,7 +4,8 @@
 
 for netdev in nba.get_netdevices():
     print(netdev)
-for coproc in nba.get_coprocessors():
+coprocessors = nba.get_coprocessors()
+for coproc in coprocessors:
     print(coproc)
 node_cpus = nba.get_cpu_node_mapping()
 for node_id, cpus in enumerate(node_cpus):
@@ -33,7 +34,7 @@
 
 coproc_threads = [
     # core_id, device_id
-    nba.CoprocThread(core_id=node_cpus[0][7] + _ht_diff, device_id=0),
+    nba.CoprocThread(core_id=node_cpus[0][7] + _ht_diff, device_id=coprocessors[0].device_id),
 ]
 
 queues = [