|
|
|
@ -305,7 +305,7 @@ class Cell:
|
|
|
|
|
logger.info("layout dict does not contain the key %s", key)
|
|
|
|
|
continue
|
|
|
|
|
if self.parameters_dict()[key].sliced:
|
|
|
|
|
logger.info("Param %s is already sliced.", key)
|
|
|
|
|
logger.debug("Param %s is already sliced.", key)
|
|
|
|
|
continue
|
|
|
|
|
layout = self.parameter_layout_dict[key]
|
|
|
|
|
new_tensor = _load_tensor_by_layout(tensor, layout)
|
|
|
|
@ -318,7 +318,7 @@ class Cell:
|
|
|
|
|
logger.info("layout dict does not contain the key %s", key)
|
|
|
|
|
continue
|
|
|
|
|
if params[key].sliced:
|
|
|
|
|
logger.info("Param %s is already sliced.", key)
|
|
|
|
|
logger.debug("Param %s is already sliced.", key)
|
|
|
|
|
continue
|
|
|
|
|
layout = self.parameter_layout_dict[key]
|
|
|
|
|
new_tensor = _load_tensor_by_layout(tensor, layout)
|
|
|
|
@ -484,7 +484,7 @@ class Cell:
|
|
|
|
|
if not auto_parallel_mode:
|
|
|
|
|
param.init_data()
|
|
|
|
|
elif param.name not in self.parameter_layout_dict:
|
|
|
|
|
logger.info("Layout dict does not contain the key %s.", param.name)
|
|
|
|
|
logger.debug("Layout dict does not contain the key %s.", param.name)
|
|
|
|
|
param.init_data(set_sliced=True)
|
|
|
|
|
else:
|
|
|
|
|
layout = self.parameter_layout_dict[param.name]
|
|
|
|
|