Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2296
Star 20931
Fork 5422
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

提交 2d461cb0 编写于 作者: Q Qiao Longfei
浏览文件
操作

code style format

上级 add4b466
显示空白变更内容
内联 并排
Showing with 38 addition and 32 deletion
python/paddle/fluid/tests/unittests/test_dist_transpiler.py
浏览文件 @ 2d461cb0
...@@ -431,7 +431,8 @@ class TestDistLookupTableBase(TranspilerTest): ...@@ -431,7 +431,8 @@ class TestDistLookupTableBase(TranspilerTest):
title_emb = emb_pool(title_ids, self.lookup_table_name, is_distributed) title_emb = emb_pool(title_ids, self.lookup_table_name, is_distributed)
brand_emb = emb_pool(brand_ids, self.lookup_table_name, is_distributed) brand_emb = emb_pool(brand_ids, self.lookup_table_name, is_distributed)
profile_emb = emb_pool(profile_ids, "profile_emb", False) profile_emb = emb_pool(profile_ids, "profile_emb", False)
fc0 = fluid.layers.concat(input=[title_emb, brand_emb, profile_emb], axis=1) fc0 = fluid.layers.concat(
input=[title_emb, brand_emb, profile_emb], axis=1)
predict = fluid.layers.fc(input=fc0, predict = fluid.layers.fc(input=fc0,
size=2, size=2,
act=None, act=None,
...@@ -471,13 +472,14 @@ class TestLocalLookupTable(TestDistLookupTableBase): ...@@ -471,13 +472,14 @@ class TestLocalLookupTable(TestDistLookupTableBase):
self.assertEqual(len(trainer.blocks), 1) self.assertEqual(len(trainer.blocks), 1)
ops = [ ops = [
'lookup_table', 'sequence_pool', 'lookup_table', 'sequence_pool', 'lookup_table', 'sequence_pool', 'lookup_table', 'sequence_pool',
'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add', 'cross_entropy', 'mean', 'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add',
'fill_constant', 'mean_grad', 'cross_entropy_grad', 'cross_entropy', 'mean', 'fill_constant', 'mean_grad',
'elementwise_add_grad', 'send', 'mul_grad', 'send', 'concat_grad', 'cross_entropy_grad', 'elementwise_add_grad', 'send', 'mul_grad',
'sequence_pool_grad', 'lookup_table_grad', 'split_selected_rows', 'send', 'send', 'concat_grad', 'sequence_pool_grad', 'lookup_table_grad',
'sequence_pool_grad', 'lookup_table_grad', 'sequence_pool_grad', 'split_selected_rows', 'send', 'sequence_pool_grad',
'lookup_table_grad', 'sum', 'split_selected_rows', 'send', 'lookup_table_grad', 'sequence_pool_grad', 'lookup_table_grad',
'send_barrier', 'recv', 'recv', 'recv', 'recv', 'fetch_barrier', 'concat', 'concat' 'sum', 'split_selected_rows', 'send', 'send_barrier', 'recv',
'recv', 'recv', 'recv', 'fetch_barrier', 'concat', 'concat'
] ]
self.assertEqual([op.type for op in trainer.blocks[0].ops], ops) self.assertEqual([op.type for op in trainer.blocks[0].ops], ops)
...@@ -510,21 +512,25 @@ class TestDistLookupTable(TestDistLookupTableBase): ...@@ -510,21 +512,25 @@ class TestDistLookupTable(TestDistLookupTableBase):
self.assertEqual(len(trainer.blocks), 1) self.assertEqual(len(trainer.blocks), 1)
ops = [ ops = [
'split_ids', 'prefetch', 'merge_ids', 'sequence_pool', 'split_ids', 'prefetch', 'merge_ids', 'sequence_pool',
'sequence_pool', 'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add', 'sequence_pool', 'lookup_table', 'sequence_pool', 'concat', 'mul',
'cross_entropy', 'mean', 'fill_constant', 'mean_grad', 'elementwise_add', 'cross_entropy', 'mean', 'fill_constant',
'cross_entropy_grad', 'elementwise_add_grad', 'send', 'mul_grad', 'mean_grad', 'cross_entropy_grad', 'elementwise_add_grad', 'send',
'send', 'concat_grad', 'sequence_pool_grad', 'lookup_table_grad', 'mul_grad', 'send', 'concat_grad', 'sequence_pool_grad',
'split_selected_rows', 'send', 'sequence_pool_grad', 'lookup_table_grad', 'lookup_table_grad', 'split_selected_rows', 'send',
'sequence_pool_grad', 'lookup_table_grad', 'sum', 'split_ids', 'send', 'send_barrier', 'sequence_pool_grad', 'lookup_table_grad', 'sequence_pool_grad',
'recv', 'recv', 'recv', 'fetch_barrier', 'concat'] 'lookup_table_grad', 'sum', 'split_ids', 'send', 'send_barrier',
'recv', 'recv', 'recv', 'fetch_barrier', 'concat'
]
self.assertEqual([op.type for op in trainer.blocks[0].ops], ops) self.assertEqual([op.type for op in trainer.blocks[0].ops], ops)
startup_ops = [ startup_ops = [
'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant',
'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant',
'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant',
'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant', 'fill_constant',
'fill_constant', 'fill_constant', 'uniform_random', 'uniform_random', 'fill_constant', 'fill_constant', 'uniform_random',
'recv', 'recv', 'recv', 'fetch_barrier', 'concat', 'fake_init'] 'uniform_random', 'recv', 'recv', 'recv', 'fetch_barrier', 'concat',
'fake_init'
]
self.assertEqual([op.type for op in trainer_startup.blocks[0].ops], self.assertEqual([op.type for op in trainer_startup.blocks[0].ops],
startup_ops) startup_ops)
...@@ -557,13 +563,12 @@ class TestAsyncLocalLookupTable(TestDistLookupTableBase): ...@@ -557,13 +563,12 @@ class TestAsyncLocalLookupTable(TestDistLookupTableBase):
'lookup_table', 'sequence_pool', 'lookup_table', 'sequence_pool', 'lookup_table', 'sequence_pool', 'lookup_table', 'sequence_pool',
'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add', 'lookup_table', 'sequence_pool', 'concat', 'mul', 'elementwise_add',
'cross_entropy', 'mean', 'fill_constant', 'mean_grad', 'cross_entropy', 'mean', 'fill_constant', 'mean_grad',
'cross_entropy_grad', 'elementwise_add_grad', 'send', 'cross_entropy_grad', 'elementwise_add_grad', 'send', 'mul_grad',
'mul_grad', 'send', 'concat_grad', 'sequence_pool_grad', 'send', 'concat_grad', 'sequence_pool_grad', 'lookup_table_grad',
'lookup_table_grad', 'split_selected_rows', 'send', 'split_selected_rows', 'send', 'sequence_pool_grad',
'sequence_pool_grad', 'lookup_table_grad', 'lookup_table_grad', 'sequence_pool_grad', 'lookup_table_grad',
'sequence_pool_grad', 'lookup_table_grad', 'sum', 'split_selected_rows', 'send', 'recv', 'recv', 'recv',
'sum', 'split_selected_rows', 'send', 'recv', 'recv', 'recv', 'recv', 'recv', 'concat', 'concat'
'concat', 'concat'
] ]
self.assertEqual([op.type for op in trainer.blocks[0].ops], ops) self.assertEqual([op.type for op in trainer.blocks[0].ops], ops)
...@@ -597,14 +602,15 @@ class TestAsyncDistLookupTable(TestDistLookupTableBase): ...@@ -597,14 +602,15 @@ class TestAsyncDistLookupTable(TestDistLookupTableBase):
self.assertEqual(len(trainer.blocks), 1) self.assertEqual(len(trainer.blocks), 1)
ops = [ ops = [
'split_ids', 'prefetch', 'merge_ids', 'sequence_pool', 'split_ids', 'prefetch', 'merge_ids', 'sequence_pool',
'sequence_pool', 'lookup_table', 'sequence_pool', 'sequence_pool', 'lookup_table', 'sequence_pool', 'concat', 'mul',
'concat', 'mul', 'elementwise_add', 'cross_entropy', 'elementwise_add', 'cross_entropy', 'mean', 'fill_constant',
'mean', 'fill_constant', 'mean_grad', 'cross_entropy_grad', 'mean_grad', 'cross_entropy_grad', 'elementwise_add_grad', 'send',
'elementwise_add_grad', 'send', 'mul_grad', 'send', 'mul_grad', 'send', 'concat_grad', 'sequence_pool_grad',
'concat_grad', 'sequence_pool_grad', 'lookup_table_grad', 'lookup_table_grad', 'split_selected_rows', 'send',
'split_selected_rows', 'send', 'sequence_pool_grad', 'sequence_pool_grad', 'lookup_table_grad', 'sequence_pool_grad',
'lookup_table_grad', 'sequence_pool_grad', 'lookup_table_grad', 'lookup_table_grad', 'sum', 'split_ids', 'send', 'recv', 'recv',
'sum', 'split_ids', 'send', 'recv', 'recv', 'recv', 'concat'] 'recv', 'concat'
]
self.assertEqual([op.type for op in trainer.blocks[0].ops], ops) self.assertEqual([op.type for op in trainer.blocks[0].ops], ops)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或