|
|
@ -486,7 +486,6 @@ def memory_optimize(input_program,
|
|
|
|
skip_opt_set = grad_set
|
|
|
|
skip_opt_set = grad_set
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
skip_opt_set.update(grad_set)
|
|
|
|
skip_opt_set.update(grad_set)
|
|
|
|
|
|
|
|
|
|
|
|
cfgs = _get_cfgs(input_program)
|
|
|
|
cfgs = _get_cfgs(input_program)
|
|
|
|
for cfg in cfgs:
|
|
|
|
for cfg in cfgs:
|
|
|
|
cfg.memory_optimize(skip_opt_set=skip_opt_set, level=level)
|
|
|
|
cfg.memory_optimize(skip_opt_set=skip_opt_set, level=level)
|
|
|
|