Skip to content

Commit 2d461cb

Browse files
jacquesqiaojacquesqiao
committed
code style format
1 parent add4b46 commit 2d461cb

File tree

1 file changed

+38
-32
lines changed

1 file changed

+38
-32
lines changed

‎python/paddle/fluid/tests/unittests/test_dist_transpiler.py

Lines changed: 38 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -431,7 +431,8 @@ def emb_pool(ids, table_name, is_distributed):
431431
title_emb = emb_pool(title_ids, self.lookup_table_name, is_distributed)
432432
brand_emb = emb_pool(brand_ids, self.lookup_table_name, is_distributed)
433433
profile_emb = emb_pool(profile_ids, "profile_emb", False)
434-
fc0 = fluid.layers.concat(input=[title_emb, brand_emb, profile_emb], axis=1)
434+
fc0 = fluid.layers.concat(
435+
input=[title_emb, brand_emb, profile_emb], axis=1)
435436
predict = fluid.layers.fc(input=fc0,
436437
size=2,
437438
act=None,
@@ -471,13 +472,14 @@ def transpiler_test_impl(self):
471472
self.assertEqual(len(trainer.blocks), 1)
472473
ops = [
473474
'lookup_table', 'sequence_pool', 'lookup_table', 'sequence_pool',
474-
'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add', 'cross_entropy', 'mean',
475-
'fill_constant', 'mean_grad', 'cross_entropy_grad',
476-
'elementwise_add_grad', 'send', 'mul_grad', 'send', 'concat_grad',
477-
'sequence_pool_grad', 'lookup_table_grad', 'split_selected_rows', 'send',
478-
'sequence_pool_grad', 'lookup_table_grad', 'sequence_pool_grad',
479-
'lookup_table_grad', 'sum', 'split_selected_rows', 'send',
480-
'send_barrier', 'recv', 'recv', 'recv', 'recv', 'fetch_barrier', 'concat', 'concat'
475+
'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add',
476+
'cross_entropy', 'mean', 'fill_constant', 'mean_grad',
477+
'cross_entropy_grad', 'elementwise_add_grad', 'send', 'mul_grad',
478+
'send', 'concat_grad', 'sequence_pool_grad', 'lookup_table_grad',
479+
'split_selected_rows', 'send', 'sequence_pool_grad',
480+
'lookup_table_grad', 'sequence_pool_grad', 'lookup_table_grad',
481+
'sum', 'split_selected_rows', 'send', 'send_barrier', 'recv',
482+
'recv', 'recv', 'recv', 'fetch_barrier', 'concat', 'concat'
481483
]
482484
self.assertEqual([op.type for op in trainer.blocks[0].ops], ops)
483485

@@ -510,21 +512,25 @@ def transpiler_test_impl(self):
510512
self.assertEqual(len(trainer.blocks), 1)
511513
ops = [
512514
'split_ids', 'prefetch', 'merge_ids', 'sequence_pool',
513-
'sequence_pool', 'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add',
514-
'cross_entropy', 'mean', 'fill_constant', 'mean_grad',
515-
'cross_entropy_grad', 'elementwise_add_grad', 'send', 'mul_grad',
516-
'send', 'concat_grad', 'sequence_pool_grad', 'lookup_table_grad',
517-
'split_selected_rows', 'send', 'sequence_pool_grad', 'lookup_table_grad',
518-
'sequence_pool_grad', 'lookup_table_grad', 'sum', 'split_ids', 'send', 'send_barrier',
519-
'recv', 'recv', 'recv', 'fetch_barrier', 'concat']
515+
'sequence_pool', 'lookup_table', 'sequence_pool', 'concat', 'mul',
516+
'elementwise_add', 'cross_entropy', 'mean', 'fill_constant',
517+
'mean_grad', 'cross_entropy_grad', 'elementwise_add_grad', 'send',
518+
'mul_grad', 'send', 'concat_grad', 'sequence_pool_grad',
519+
'lookup_table_grad', 'split_selected_rows', 'send',
520+
'sequence_pool_grad', 'lookup_table_grad', 'sequence_pool_grad',
521+
'lookup_table_grad', 'sum', 'split_ids', 'send', 'send_barrier',
522+
'recv', 'recv', 'recv', 'fetch_barrier', 'concat'
523+
]
520524
self.assertEqual([op.type for op in trainer.blocks[0].ops], ops)
521525
startup_ops = [
522526
'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant',
523527
'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant',
524528
'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant',
525529
'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant',
526-
'fill_constant', 'fill_constant', 'uniform_random', 'uniform_random',
527-
'recv', 'recv', 'recv', 'fetch_barrier', 'concat', 'fake_init']
530+
'fill_constant', 'fill_constant', 'uniform_random',
531+
'uniform_random', 'recv', 'recv', 'recv', 'fetch_barrier', 'concat',
532+
'fake_init'
533+
]
528534
self.assertEqual([op.type for op in trainer_startup.blocks[0].ops],
529535
startup_ops)
530536

@@ -557,13 +563,12 @@ def transpiler_test_impl(self):
557563
'lookup_table', 'sequence_pool', 'lookup_table', 'sequence_pool',
558564
'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add',
559565
'cross_entropy', 'mean', 'fill_constant', 'mean_grad',
560-
'cross_entropy_grad', 'elementwise_add_grad', 'send',
561-
'mul_grad', 'send', 'concat_grad', 'sequence_pool_grad',
562-
'lookup_table_grad', 'split_selected_rows', 'send',
563-
'sequence_pool_grad', 'lookup_table_grad',
564-
'sequence_pool_grad', 'lookup_table_grad',
565-
'sum', 'split_selected_rows', 'send', 'recv', 'recv', 'recv', 'recv',
566-
'concat', 'concat'
566+
'cross_entropy_grad', 'elementwise_add_grad', 'send', 'mul_grad',
567+
'send', 'concat_grad', 'sequence_pool_grad', 'lookup_table_grad',
568+
'split_selected_rows', 'send', 'sequence_pool_grad',
569+
'lookup_table_grad', 'sequence_pool_grad', 'lookup_table_grad',
570+
'sum', 'split_selected_rows', 'send', 'recv', 'recv', 'recv',
571+
'recv', 'concat', 'concat'
567572
]
568573
self.assertEqual([op.type for op in trainer.blocks[0].ops], ops)
569574

@@ -597,14 +602,15 @@ def transpiler_test_impl(self):
597602
self.assertEqual(len(trainer.blocks), 1)
598603
ops = [
599604
'split_ids', 'prefetch', 'merge_ids', 'sequence_pool',
600-
'sequence_pool', 'lookup_table', 'sequence_pool',
601-
'concat', 'mul', 'elementwise_add', 'cross_entropy',
602-
'mean', 'fill_constant', 'mean_grad', 'cross_entropy_grad',
603-
'elementwise_add_grad', 'send', 'mul_grad', 'send',
604-
'concat_grad', 'sequence_pool_grad', 'lookup_table_grad',
605-
'split_selected_rows', 'send', 'sequence_pool_grad',
606-
'lookup_table_grad', 'sequence_pool_grad', 'lookup_table_grad',
607-
'sum', 'split_ids', 'send', 'recv', 'recv', 'recv', 'concat']
605+
'sequence_pool', 'lookup_table', 'sequence_pool', 'concat', 'mul',
606+
'elementwise_add', 'cross_entropy', 'mean', 'fill_constant',
607+
'mean_grad', 'cross_entropy_grad', 'elementwise_add_grad', 'send',
608+
'mul_grad', 'send', 'concat_grad', 'sequence_pool_grad',
609+
'lookup_table_grad', 'split_selected_rows', 'send',
610+
'sequence_pool_grad', 'lookup_table_grad', 'sequence_pool_grad',
611+
'lookup_table_grad', 'sum', 'split_ids', 'send', 'recv', 'recv',
612+
'recv', 'concat'
613+
]
608614
self.assertEqual([op.type for op in trainer.blocks[0].ops], ops)
609615

610616

0 commit comments

Comments
 (0)