@@ -56,12 +56,13 @@ def to_label_decomposed_graph(graph):
5656 alloc_r << alloc .T
5757 print ("Boolean matrix for alloc_r nvals: " , alloc_r .nvals )
5858
59- print ("mask start" )
60- mask_v = Vector (BOOL , graph .ncols , name = "mask_vector" )
61- mask_v (op .lor ) << alloc .reduce_columnwise ("lor" )
62- mask_v (op .lor ) << alloc .reduce_rowwise ("lor" )
59+ #print("mask start")
60+ #mask_v = Vector(BOOL, graph.ncols, name = "mask_vector")
61+ #mask_v(op.lor) << alloc.reduce_columnwise("lor")
62+ #mask_v(op.lor) << alloc.reduce_rowwise("lor")
63+
64+ #print("entrypoints start")
6365
64- print ("entrypoints start" )
6566 #entrypoints = Vector(bool,graph.nrows, name="entrypoints")
6667 #entrypoints << graph.reduce_columnwise(op.lor)
6768 #mask_v(op.lor) << Vector.from_coo(list(set(range(0,graph.nrows)).difference(entrypoints.to_coo(values=False)[0])), values=True, dtype = BOOL)
@@ -74,11 +75,11 @@ def to_label_decomposed_graph(graph):
7475 store_i << graph .select (graphblas .select .select_store ).apply (graphblas .unary .decode_store )
7576 print ("Matrix for store_i nvals: " , store_i .nvals )
7677
77- mask_v (op .lor ) << load_i .reduce_columnwise ("lor" )
78- mask_v (op .lor ) << load_i .reduce_rowwise ("lor" )
78+ # mask_v(op.lor) << load_i.reduce_columnwise("lor")
79+ # mask_v(op.lor) << load_i.reduce_rowwise("lor")
7980
80- mask_v (op .lor ) << store_i .reduce_columnwise ("lor" )
81- mask_v (op .lor ) << store_i .reduce_rowwise ("lor" )
81+ # mask_v(op.lor) << store_i.reduce_columnwise("lor")
82+ # mask_v(op.lor) << store_i.reduce_rowwise("lor")
8283
8384 store_block_count = store_i .reduce_scalar ("max" ).get (0 ) + 1
8485 load_block_count = load_i .reduce_scalar ("max" ).get (0 ) + 1
@@ -106,7 +107,7 @@ def to_label_decomposed_graph(graph):
106107
107108
108109
109- assign_mask = mask_v .diag (name = "assign_mask" )
110+ # assign_mask = mask_v.diag(name = "assign_mask")
110111
111112 assign = Matrix (BOOL , graph .ncols , graph .nrows , name = "assign_after_intersection" )
112113 assign << graph .select (graphblas .select .select_assign )
@@ -133,17 +134,6 @@ def to_label_decomposed_graph(graph):
133134
134135 #print_matrix_to_dot(assign_r,"assign_r.dot")
135136
136-
137-
138-
139-
140-
141-
142-
143-
144-
145-
146-
147137 matrices : Dict [Symbol , Matrix ] = {}
148138
149139 matrices [Symbol ('alloc' )] = alloc
@@ -164,10 +154,10 @@ def to_label_decomposed_graph(graph):
164154 matrices = matrices
165155 )
166156
167- def add_context (file_path ):
157+ def add_context (file_path , max_num_of_contexts ):
168158 load_graph_start = time .perf_counter ()
169159
170- graph ,number_of_contexts = load_graph (file_path )
160+ graph ,number_of_contexts = load_graph (file_path , max_num_of_contexts )
171161
172162 load_graph_end = time .perf_counter ()
173163 print ("Graph loaded in " , load_graph_end - load_graph_start )
0 commit comments