@@ -63,6 +63,15 @@ if(WITH_DISTRIBUTE AND WITH_GPU)
63
63
py_test_modules (test_engine_callbacks MODULES test_engine_callbacks )
64
64
set_tests_properties (test_engine_callbacks
65
65
PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE" TIMEOUT 50 )
66
+ py_test_modules (test_parallel_tuner MODULES test_parallel_tuner ENVS
67
+ ${dist_ENVS} )
68
+ set_tests_properties (test_parallel_tuner PROPERTIES TIMEOUT 120 )
69
+ py_test_modules (test_parallel_tuner_full MODULES test_parallel_tuner_full
70
+ ENVS ${dist_ENVS} )
71
+ set_tests_properties (test_parallel_tuner_full PROPERTIES TIMEOUT 120 )
72
+ py_test_modules (test_parallel_tuner_predict MODULES
73
+ test_parallel_tuner_predict ENVS ${dist_ENVS} )
74
+ set_tests_properties (test_parallel_tuner_predict PROPERTIES TIMEOUT 120 )
66
75
67
76
py_test_modules (test_while_op_completion MODULES test_while_op_completion
68
77
ENVS ${dist_ENVS} )
@@ -90,6 +99,7 @@ if(WITH_DISTRIBUTE AND WITH_GPU)
90
99
py_test_modules (test_prim_dist_op MODULES test_prim_dist_op ENVS ${dist_ENVS} )
91
100
py_test_modules (test_to_static MODULES test_to_static ENVS ${dist_ENVS} )
92
101
py_test_modules (test_dist_op_cost MODULES test_dist_op_cost ENVS ${dist_ENVS} )
102
+
93
103
py_test_modules (test_cluster_v2 MODULES test_cluster_v2 )
94
104
py_test_modules (test_process_mesh_v2 MODULES test_process_mesh_v2 )
95
105
py_test_modules (test_dist_attr_v2 MODULES test_dist_attr_v2 )
@@ -99,20 +109,10 @@ if(WITH_DISTRIBUTE AND WITH_GPU)
99
109
py_test_modules (test_interface MODULES test_interface )
100
110
py_test_modules (test_strategy MODULES test_strategy )
101
111
py_test_modules (test_pass_quantization MODULES test_pass_quantization )
102
-
103
112
py_test_modules (test_dist_shape MODULES test_dist_shape )
104
113
py_test_modules (test_dist_assign MODULES test_dist_assign )
105
114
py_test_modules (test_conditional_block_reshard MODULES
106
115
test_conditional_block_reshard )
107
-
108
- py_test_modules (test_parallel_tuner MODULES test_parallel_tuner ENVS
109
- ${dist_ENVS} )
110
- set_tests_properties (test_parallel_tuner PROPERTIES TIMEOUT 120 )
111
- py_test_modules (test_parallel_tuner_full MODULES test_parallel_tuner_full
112
- ENVS ${dist_ENVS} )
113
- set_tests_properties (test_parallel_tuner_full PROPERTIES TIMEOUT 120 )
114
- py_test_modules (test_parallel_tuner_predict MODULES
115
- test_parallel_tuner_predict ENVS ${dist_ENVS} )
116
- set_tests_properties (test_parallel_tuner_predict PROPERTIES TIMEOUT 120 )
116
+ py_test_modules (test_engine_api_error MODULES test_engine_api_error )
117
117
118
118
endif ()
0 commit comments