diff --git a/graphengine b/graphengine index 63a27119bf7..86596e18efb 160000 --- a/graphengine +++ b/graphengine @@ -1 +1 @@ -Subproject commit 63a27119bf705511946f690c55c722c356f4b397 +Subproject commit 86596e18efba333aed421fcc7aa453aa57de001a diff --git a/mindspore/ops/_op_impl/tbe/tile.py b/mindspore/ops/_op_impl/tbe/tile.py index d55d427266e..178a623091f 100644 --- a/mindspore/ops/_op_impl/tbe/tile.py +++ b/mindspore/ops/_op_impl/tbe/tile.py @@ -26,9 +26,8 @@ tile_op_info = TBERegOp("Tile") \ .attr("multiples", "optional", "listInt", "all")\ .input(0, "x1", False, "required", "all") \ .output(0, "y", False, "required", "all") \ - .dtype_format(DataType.I32_Default, DataType.I32_Default) \ - .dtype_format(DataType.I16_Default, DataType.I16_Default) \ - .dtype_format(DataType.F32_Default, DataType.F32_Default) \ + .op_pattern("dynamicFormat") \ + .dtype_format(DataType.None_None, DataType.None_None) \ .get_op_info() diff --git a/mindspore/train/model.py b/mindspore/train/model.py index dd2fdfab670..32da5a9e012 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -548,8 +548,6 @@ class Model: >>> model.train(2, dataset) """ check_bool(dataset_sink_mode) - if sink_size == -1: - sink_size = train_dataset.get_dataset_size() check_int(sink_size) if sink_size < -1 or sink_size == 0: raise ValueError("The sink_size must be -1 or positive, but got sink_size {}.".format(sink_size)) diff --git a/tests/dataset_mock.py b/tests/dataset_mock.py index ee9af3c40f3..0ee0a90813c 100644 --- a/tests/dataset_mock.py +++ b/tests/dataset_mock.py @@ -21,7 +21,7 @@ from mindspore import Tensor class MindData: """ Stub for MindData """ - def __init__(self, size=1, batch_size=None, repeat_count=1, + def __init__(self, size=None, batch_size=None, repeat_count=1, np_types=None, output_shapes=None, input_indexs=()): self._size = size self._batch_size = batch_size diff --git a/tests/ut/python/parallel/test_loss_scale.py b/tests/ut/python/parallel/test_loss_scale.py index 9649679474d..88997160f8e 100644 --- a/tests/ut/python/parallel/test_loss_scale.py +++ b/tests/ut/python/parallel/test_loss_scale.py @@ -113,7 +113,7 @@ class TrainOneStepWithLossScaleCell(nn.Cell): class DatasetLenet(MindData): def __init__(self, predict, label, length=3): - super(DatasetLenet, self).__init__(size=length) + super(DatasetLenet, self).__init__() self.predict = predict self.label = label self.index = 0