@@ -138,43 +138,6 @@ def test_static_graph(self):
138
138
139
139
paddle .disable_static ()
140
140
141
- def test_static_graph_2 (self ):
142
- """Test static graph execution"""
143
- np .random .seed (114514 )
144
- axis = paddle .to_tensor (- 1 )
145
- paddle .enable_static ()
146
- with paddle .static .program_guard (paddle .static .Program ()):
147
- x = paddle .static .data (name = 'x' , shape = [None , 9 ], dtype = 'float32' )
148
- result0 , result1 , result2 = split (x , 4 , dim = axis )
149
- output = result0 + result1 * result2
150
-
151
- place = (
152
- paddle .CUDAPlace (0 )
153
- if paddle .is_compiled_with_cuda ()
154
- else paddle .CPUPlace ()
155
- )
156
- exe = paddle .static .Executor (place )
157
-
158
- input_data = np .random .rand (3 , 9 ).astype ('float32' )
159
- feed = {'x' : input_data }
160
-
161
- results = exe .run (
162
- feed = feed , fetch_list = [result0 , result1 , result2 , output ]
163
- )
164
-
165
- np .testing .assert_allclose (results [0 ], input_data [:, 0 :4 ])
166
- np .testing .assert_allclose (results [1 ], input_data [:, 4 :8 ])
167
- np .testing .assert_allclose (results [2 ], input_data [:, 8 :9 ])
168
-
169
- expected_output = (
170
- input_data [:, 0 :4 ] + input_data [:, 4 :8 ] * input_data [:, - 1 :]
171
- )
172
- np .testing .assert_allclose (
173
- expected_output , results [3 ], rtol = 1e-4 , atol = 1e-4
174
- )
175
-
176
- paddle .disable_static ()
177
-
178
141
def test_error_hint (self ):
179
142
"""Test whether there will be correct exception when users pass paddle.split kwargs in paddle.compat.split, vice versa."""
180
143
0 commit comments