File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
utensor_cgen/backend/utensor/code_generator/rearch Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -352,7 +352,7 @@ def _compute_meta_data_size(self, ugraph):
352352 # TODO: if mem_optimizer is None, use a default mem optimizer
353353 if self .meta_data_pool_size == 'auto' :
354354 # TODO: compute actual meta data size with ugraph
355- size = 256
355+ size = 2048
356356 else :
357357 size = self .meta_data_pool_size
358358 return size
@@ -362,7 +362,7 @@ def _compute_ram_data_size(self, ugraph):
362362 if self .ram_data_pool_size == 'auto' :
363363 # TODO: compute actual ram data size with ugraph
364364 if '_tensor_alloc' in ugraph .attributes :
365- size = ugraph .attributes ['_tensor_alloc' ].total_size + 500
365+ size = ugraph .attributes ['_tensor_alloc' ].total_size + 3000
366366 else :
367367 size = 256
368368 else :
You can’t perform that action at this time.
0 commit comments