Skip to content

Commit 5d030fb

Browse files
committed
minor params fix
1 parent 9a1d020 commit 5d030fb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

utensor_cgen/backend/utensor/code_generator/rearch/_code_generator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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:

0 commit comments

Comments
 (0)