Skip to content

Commit 75b3605

Browse files
committed
add test_semi_auto_parallel_c_cross_entropy to whitelist
1 parent 2935895 commit 75b3605

3 files changed

Lines changed: 7 additions & 1 deletion

File tree

test/auto_parallel/hybrid_strategy/test_semi_auto_parallel_c_cross_entropy.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,11 @@ def test_mp(self):
3535
"semi_auto_parallel_c_cross_entropy_mp.py",
3636
)
3737

38+
def test_mp_pir(self):
39+
os.environ["FLAGS_enable_pir_in_executor"] = "True"
40+
self.test_mp()
41+
os.environ["FLAGS_enable_pir_in_executor"] = "False"
42+
3843

3944
class TestParallelCrossEntropyHybrid(test_base.CommunicationTestDistBase):
4045
def setUp(self):

test/white_list/pir_op_test_white_list

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -267,6 +267,7 @@ test_seed_op
267267
test_segment_ops
268268
test_segment_ops_static_build
269269
test_selu_op
270+
test_semi_auto_parallel_c_cross_entropy
270271
test_sgd_op
271272
test_shape_mkldnn_op
272273
test_shape_op

0 commit comments

Comments
 (0)