m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix parallel timeout

Browse Source
pull/14606/head
yao_yf 4 years ago
parent 5298b1d225
commit a83fb3316b
32 changed files with 45 additions and 45 deletions
Show Stats Download Patch File Download Diff File

4
tests/ut/python/parallel/test_auto_parallel_double_subgraphs.py
Unescape View File

@ -105,7 +105,7 @@ class TrainStepWarp(nn.Cell):
def test_double_subgraphs():
_set_multi_subgraphs()
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(device_num=8, global_rank=0)
context.set_auto_parallel_context(parallel_mode="auto_parallel")
net = TrainStepWarp(NetWithLoss(Net()))
@ -156,7 +156,7 @@ class DatasetLenet():
return self
def test_double_subgraphs_train():
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(device_num=1, global_rank=0)
context.set_auto_parallel_context(parallel_mode="auto_parallel")
net = TrainStepWarp(NetWithLoss(Net()))

2
tests/ut/python/parallel/test_auto_parallel_for_loop.py
Unescape View File

@ -118,7 +118,7 @@ _w1 = Tensor(np.ones([512, 128]), dtype=ms.float32)
def test_auto_parallel():
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=16, global_rank=0)
net = Full(_w1, 3)
net.set_auto_parallel()

2
tests/ut/python/parallel/test_auto_parallel_for_loop_multi_subgraph.py
Unescape View File

@ -121,7 +121,7 @@ class TrainStepWarp(nn.Cell):
def test_double_subgraphs():
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=8, global_rank=0)
net = TrainStepWarp(NetWithLoss(Net()))
_set_multi_subgraphs()

2
tests/ut/python/parallel/test_auto_parallel_for_loop_reshape.py
Unescape View File

@ -125,7 +125,7 @@ _w1 = Tensor(np.ones([512, 128, 1]), dtype=ms.float32)
def test_auto_parallel():
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=16, global_rank=0)
net = Full(_w1, 3)
net.set_auto_parallel()

2
tests/ut/python/parallel/test_auto_parallel_for_loop_simplify.py
Unescape View File

@ -83,7 +83,7 @@ _w1 = Tensor(np.ones([512, 128]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
learning_rate = 0.1
momentum = 0.9
epoch_size = 2

2
tests/ut/python/parallel/test_auto_parallel_two_bn.py
Unescape View File

@ -68,7 +68,7 @@ def test_two_bn():
out = self.block2(out)
return out
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(device_num=8, global_rank=0)
context.set_auto_parallel_context(parallel_mode="auto_parallel")
net = NetWithLoss(Net())

4
tests/ut/python/parallel/test_broadcast_to.py
Unescape View File

@ -64,7 +64,7 @@ _x2 = Tensor(np.ones([64, 64]), dtype=ms.float32)
def compile_net(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()
@ -74,7 +74,7 @@ def compile_net(net):
def compile_net2(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()

2
tests/ut/python/parallel/test_concat.py
Unescape View File

@ -80,7 +80,7 @@ w3 = Tensor(np.ones([64, 64, 32]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()

2
tests/ut/python/parallel/test_embeddinglookup.py
Unescape View File

@ -95,7 +95,7 @@ def test_embeddinglookup_reducescatter_false_grad():
def test_embeddinglookup_reducescatter_true_grad():
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
shape = [8, 8]
offset = 8
net = GradWrap(NetWithLoss(Net(shape, offset)))

4
tests/ut/python/parallel/test_eval.py
Unescape View File

@ -52,7 +52,7 @@ _b = Tensor(np.ones([64, 64]), dtype=ms.float32)
def test_train_and_eval():
context.set_context(save_graphs=True, mode=0)
context.set_context(save_graphs=False, mode=0)
context.set_auto_parallel_context(parallel_mode="semi_auto_parallel", device_num=16)
strategy1 = ((4, 4), (4, 4))
strategy2 = ((4, 4),)
@ -69,7 +69,7 @@ def test_train_and_eval():
context.reset_auto_parallel_context()
def test_train_and_eval_auto():
context.set_context(save_graphs=True, mode=0)
context.set_context(save_graphs=False, mode=0)
context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=16)
strategy1 = ((4, 4), (4, 4))
strategy2 = ((4, 4),)

2
tests/ut/python/parallel/test_loss_and_o2_level.py
Unescape View File

@ -63,7 +63,7 @@ _w1 = Tensor(np.ones([512, 128]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
learning_rate = 0.1
momentum = 0.9
epoch_size = 2

2
tests/ut/python/parallel/test_loss_scale.py
Unescape View File

@ -194,7 +194,7 @@ def test_loss_scale():
def test_loss_scale2():
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
context.set_auto_parallel_context(parallel_mode=ParallelMode.SEMI_AUTO_PARALLEL, device_num=8)
predict = Tensor(np.ones([64, 64]), dtype=ms.float32)
label = Tensor(np.ones([64,]), dtype=ms.int32)

6
tests/ut/python/parallel/test_manual_embedding_lookup.py
Unescape View File

@ -66,7 +66,7 @@ _b = Tensor(np.ones([8, 8, 8]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
optimizer = LazyAdam(net.trainable_params(), learning_rate=0.1)
optimizer.sparse_opt.add_prim_attr("primitive_target", "CPU")
train_net = TrainOneStepCell(net, optimizer)
@ -113,7 +113,7 @@ def test_normal_split_with_offset():
def test_auto_parallel_error():
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=2, global_rank=0)
net = Net()
with pytest.raises(RuntimeError):
@ -121,7 +121,7 @@ def test_auto_parallel_error():
def test_auto_parallel():
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=2, global_rank=0)
net = Net(split_string="fake")
compile_net(net)

4
tests/ut/python/parallel/test_manual_gatherv2.py
Unescape View File

@ -60,7 +60,7 @@ _b = Tensor(np.ones([64, 8]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()
@ -106,7 +106,7 @@ def test_normal_split_with_offset():
def test_auto_parallel_error():
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
context.set_auto_parallel_context(parallel_mode="auto_parallel", device_num=2, global_rank=0)
net = Net()
with pytest.raises(RuntimeError):

2
tests/ut/python/parallel/test_model_with_loss.py
Unescape View File

@ -63,7 +63,7 @@ _w1 = Tensor(np.ones([512, 128]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
learning_rate = 0.1
momentum = 0.9
epoch_size = 2

2
tests/ut/python/parallel/test_model_without_loss.py
Unescape View File

@ -103,7 +103,7 @@ w3 = Tensor(np.ones([64, 64, 32]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
learning_rate = 0.1
momentum = 0.9
epoch_size = 2

2
tests/ut/python/parallel/test_o2_level.py
Unescape View File

@ -90,7 +90,7 @@ _w2 = Tensor(np.ones([128, 64, 1]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
learning_rate = 0.1
momentum = 0.9
epoch_size = 2

8
tests/ut/python/parallel/test_pack.py
Unescape View File

@ -136,7 +136,7 @@ _x_c = Tensor(np.ones([8, 8, 8]), dtype=ms.float32)
def compile_net(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()
@ -146,7 +146,7 @@ def compile_net(net):
def compile_net1(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()
@ -156,7 +156,7 @@ def compile_net1(net):
def compile_net2(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()
@ -166,7 +166,7 @@ def compile_net2(net):
def compile_net_con(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()

2
tests/ut/python/parallel/test_parameter_merge.py
Unescape View File

@ -79,7 +79,7 @@ def clean_all_ckpt_files(folder_path):
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
learning_rate = 0.1
momentum = 0.9
epoch_size = 2

2
tests/ut/python/parallel/test_range.py
Unescape View File

@ -74,7 +74,7 @@ _w1 = Tensor(np.ones([64, 8]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
learning_rate = 0.1
momentum = 0.9
epoch_size = 2

2
tests/ut/python/parallel/test_reluv2.py
Unescape View File

@ -40,7 +40,7 @@ _x = Tensor(np.ones([32, 16, 48, 64]), dtype=ms.float32)
def compile_net(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()

4
tests/ut/python/parallel/test_repeated_calc.py
Unescape View File

@ -73,7 +73,7 @@ def test_tensoradd_reshape_matmul():
strategy2 = ((8, 1), (1, 8))
net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
x = Tensor(np.ones([32, 8, 16]), dtype=ms.float32)
y = Tensor(np.ones([32, 8, 16]), dtype=ms.float32)
@ -99,7 +99,7 @@ def test_two_matmul():
strategy2 = ((8, 1), (1, 1))
net = GradWrap(NetWithLoss(Net(strategy1, strategy2)))
context.set_auto_parallel_context(parallel_mode="semi_auto_parallel")
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
x = Tensor(np.ones([128, 32]), dtype=ms.float32)
y = Tensor(np.ones([32, 64]), dtype=ms.float32)

2
tests/ut/python/parallel/test_reshape_optimized.py
Unescape View File

@ -40,7 +40,7 @@ _b = Tensor(np.ones([128, 64, 32]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()

2
tests/ut/python/parallel/test_reshape_skip_redistribution.py
Unescape View File

@ -43,7 +43,7 @@ _x = Tensor(np.ones([64, 64]), dtype=ms.float32)
_b = Tensor(np.ones([128, 64, 32]), dtype=ms.float32)
def compile_net(net):
context.set_context(save_graphs=True)
context.set_context(save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()

4
tests/ut/python/parallel/test_split.py
Unescape View File

@ -75,7 +75,7 @@ _x1 = Tensor(np.ones([48, 64, 32]), dtype=ms.float32)
_w2 = Tensor(np.ones([48, 64, 32]), dtype=ms.float32)
def compile_net(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()
@ -85,7 +85,7 @@ def compile_net(net):
def compile_net1(net):
context.set_context(mode=context.GRAPH_MODE, save_graphs=True)
context.set_context(mode=context.GRAPH_MODE, save_graphs=False)
optimizer = Momentum(net.trainable_params(), learning_rate=0.1, momentum=0.9)
train_net = TrainOneStepCell(net, optimizer)
train_net.set_auto_parallel()

Some files were not shown because too many files have changed in this diff Show More

Write Preview
Loading…
Cancel
Save