瀏覽代碼

profiling: use starpu_memory_nodes_get_count() instead of STARPU_MAX_NODES

Samuel Pitoiset 9 年之前
父節點
當前提交
3d2242f744
共有 1 個文件被更改,包括 5 次插入3 次删除
  1. 5 3
      src/profiling/profiling.c

+ 5 - 3
src/profiling/profiling.c

@@ -364,10 +364,12 @@ void _starpu_profiling_set_task_push_end_time(struct starpu_task *task)
 
 void _starpu_initialize_busid_matrix(void)
 {
+	int num_nodes = starpu_memory_nodes_get_count();
 	int i, j;
-	for (j = 0; j < STARPU_MAXNODES; j++)
-	for (i = 0; i < STARPU_MAXNODES; i++)
-		busid_matrix[i][j] = -1;
+
+	for (j = 0; j <= num_nodes; j++)
+		for (i = 0; i <= num_nodes; i++)
+			busid_matrix[i][j] = -1;
 
 	busid_cnt = 0;
 }