fix several windows UTs in PIR (#68191)

rsmallblue · web-flow · commit 5b5eae1aa7b8 · 2024-09-18T14:08:11.000+08:00
diff --git a/test/ir/inference/test_adaptive_pool2d_convert_global_pass_autoscan.py b/test/ir/inference/test_adaptive_pool2d_convert_global_pass_autoscan.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import sys
 import unittest
 
 import hypothesis.strategies as st
@@ -92,11 +93,16 @@ def sample_predictor_configs(self, program_config):
         yield config, ['pool2d'], (1e-5, 1e-5)
 
     def test(self):
+        max_example = 300
+        min_success_num = 40
+        if sys.platform == 'win32':
+            max_example = 10
+            min_success_num = 4
         self.run_and_statis(
             quant=False,
-            max_examples=300,
+            max_examples=max_example,
             passes=["adaptive_pool2d_convert_global_pass"],
-            min_success_num=40,
+            min_success_num=min_success_num,
         )
 
 
diff --git a/test/ir/inference/test_reverse_roll_fuse_pass.py b/test/ir/inference/test_reverse_roll_fuse_pass.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import math
+import sys
 import unittest
 from functools import partial
 
@@ -208,12 +209,17 @@ def generate_input(attrs):
         return program_config
 
     def test(self):
+        max_examples = 50
+        min_success_num = 50
+        if sys.platform == "win32":
+            max_examples = 5
+            min_success_num = 5
         self.run_and_statis(
             quant=False,
-            max_examples=50,
+            max_examples=max_examples,
             passes=["reverse_roll_fuse_pass"],
             max_duration=250,
-            min_success_num=50,
+            min_success_num=min_success_num,
         )
 
 
@@ -385,12 +391,17 @@ def generate_input(attrs):
         return program_config
 
     def test(self):
+        max_examples = 50
+        min_success_num = 50
+        if sys.platform == "win32":
+            max_examples = 5
+            min_success_num = 5
         self.run_and_statis(
             quant=False,
-            max_examples=50,
+            max_examples=max_examples,
             passes=["reverse_roll_fuse_pass"],
             max_duration=250,
-            min_success_num=50,
+            min_success_num=min_success_num,
         )
 
 
diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt
@@ -1161,7 +1161,7 @@ set_tests_properties(test_elementwise_mul_op PROPERTIES TIMEOUT 120)
 set_tests_properties(test_fractional_max_pool2d_op PROPERTIES TIMEOUT 120)
 set_tests_properties(test_graph_send_ue_recv_op PROPERTIES TIMEOUT 60)
 set_tests_properties(test_graph_send_uv_op PROPERTIES TIMEOUT 60)
-set_tests_properties(test_gru_op PROPERTIES TIMEOUT 200)
+set_tests_properties(test_gru_op PROPERTIES TIMEOUT 300)
 set_tests_properties(test_gru_unit_op PROPERTIES TIMEOUT 120)
 set_tests_properties(test_imperative_star_gan_with_gradient_penalty
                      PROPERTIES TIMEOUT 120)
diff --git a/test/legacy_test/test_audio_functions.py b/test/legacy_test/test_audio_functions.py
@@ -66,17 +66,17 @@ def test_audio_function(self, val: float, htk_flag: bool):
         mel_paddle_tensor = paddle.audio.functional.hz_to_mel(
             paddle.to_tensor([val]), htk_flag
         )
-        mel_librosa = librosa.hz_to_mel(val, htk_flag)
+        mel_librosa = librosa.hz_to_mel(val, htk=htk_flag)
         np.testing.assert_almost_equal(mel_paddle, mel_librosa, decimal=5)
         np.testing.assert_almost_equal(
-            mel_paddle_tensor.numpy(), mel_librosa, decimal=4
+            mel_paddle_tensor.numpy(), mel_librosa, decimal=3
         )
 
         hz_paddle = paddle.audio.functional.mel_to_hz(val, htk_flag)
         hz_paddle_tensor = paddle.audio.functional.mel_to_hz(
             paddle.to_tensor([val]), htk_flag
         )
-        hz_librosa = librosa.mel_to_hz(val, htk_flag)
+        hz_librosa = librosa.mel_to_hz(val, htk=htk_flag)
         np.testing.assert_almost_equal(hz_paddle, hz_librosa, decimal=4)
         np.testing.assert_almost_equal(
             hz_paddle_tensor.numpy(), hz_librosa, decimal=4
@@ -123,12 +123,12 @@ def test_audio_function_static(self, val: float, htk_flag: bool):
                 ],
             )
 
-            mel_librosa = librosa.hz_to_mel(val, htk_flag)
+            mel_librosa = librosa.hz_to_mel(val, htk=htk_flag)
             np.testing.assert_almost_equal(
-                mel_paddle_tensor_ret, mel_librosa, decimal=4
+                mel_paddle_tensor_ret, mel_librosa, decimal=3
             )
 
-            hz_librosa = librosa.mel_to_hz(val, htk_flag)
+            hz_librosa = librosa.mel_to_hz(val, htk=htk_flag)
             np.testing.assert_almost_equal(
                 hz_paddle_tensor_ret, hz_librosa, decimal=4
             )
@@ -147,7 +147,7 @@ def test_audio_function_mel(
         self, n_mels: int, f_min: float, f_max: float, htk_flag: bool
     ):
         librosa_mel_freq = librosa.mel_frequencies(
-            n_mels, f_min, f_max, htk_flag
+            n_mels, fmin=f_min, fmax=f_max, htk=htk_flag
         )
         paddle_mel_freq = paddle.audio.functional.mel_frequencies(
             n_mels, f_min, f_max, htk_flag, 'float64'
@@ -175,7 +175,7 @@ def test_audio_function_mel_static(
         exe = paddle.static.Executor()
         (paddle_mel_freq_ret,) = exe.run(main, fetch_list=[paddle_mel_freq])
         librosa_mel_freq = librosa.mel_frequencies(
-            n_mels, f_min, f_max, htk_flag
+            n_mels, fmin=f_min, fmax=f_max, htk=htk_flag
         )
         np.testing.assert_almost_equal(
             paddle_mel_freq_ret, librosa_mel_freq, decimal=3
@@ -185,7 +185,7 @@ def test_audio_function_mel_static(
 
     @parameterize([8000, 16000], [64, 128, 256])
     def test_audio_function_fft(self, sr: int, n_fft: int):
-        librosa_fft = librosa.fft_frequencies(sr, n_fft)
+        librosa_fft = librosa.fft_frequencies(sr=sr, n_fft=n_fft)
         paddle_fft = paddle.audio.functional.fft_frequencies(sr, n_fft)
         np.testing.assert_almost_equal(paddle_fft, librosa_fft, decimal=5)
 
diff --git a/test/legacy_test/test_translated_layer.py b/test/legacy_test/test_translated_layer.py
@@ -196,7 +196,7 @@ def test_get_output_spec(self):
             paddle.static.InputSpec(
                 shape=[None, CLASS_NUM],
                 dtype='float32',
-                name='translated_layer/scale_0.tmp_1',
+                name='output_0',
             )
         ]
         actual_spec = translated_layer._output_spec()

Original file line number	Diff line number	Diff line change
`@@ -196,7 +196,7 @@ def test_get_output_spec(self):`
`196`	`196`	`paddle.static.InputSpec(`
`197`	`197`	`shape=[None, CLASS_NUM],`
`198`	`198`	`dtype='float32',`
`199`		`- name='translated_layer/scale_0.tmp_1',`
	`199`	`+ name='output_0',`
`200`	`200`	`)`
`201`	`201`	`]`
`202`	`202`	`actual_spec = translated_layer._output_spec()`