GitOrigin-RevId: e3d03b4d1d
tags/v1.9.0
| @@ -507,7 +507,7 @@ def test_interpolate_fastpath(): | |||||
| [(1, 1, 10, 10), (5, 5)], | [(1, 1, 10, 10), (5, 5)], | ||||
| [(1, 3, 10, 10), (20, 20)], | [(1, 3, 10, 10), (20, 20)], | ||||
| [(10, 1, 10, 10), (1, 1)], | [(10, 1, 10, 10), (1, 1)], | ||||
| # [(10, 10, 1, 1), (10, 10)], # FIXME, it causes random CI failure | |||||
| [(10, 10, 1, 1), (10, 10)], | |||||
| ] | ] | ||||
| for inp_shape, target_shape in test_cases: | for inp_shape, target_shape in test_cases: | ||||
| x = tensor(np.random.randn(*inp_shape), dtype=np.float32) | x = tensor(np.random.randn(*inp_shape), dtype=np.float32) | ||||
| @@ -39,6 +39,8 @@ from megengine.random import uniform | |||||
| get_device_count("xpu") <= 2, reason="xpu counts need > 2", | get_device_count("xpu") <= 2, reason="xpu counts need > 2", | ||||
| ) | ) | ||||
| def test_gaussian_op(): | def test_gaussian_op(): | ||||
| # FIXME: remove this sync | |||||
| mge.core.set_option("async_level", 0) | |||||
| set_global_seed(1024) | set_global_seed(1024) | ||||
| shape = ( | shape = ( | ||||
| 8, | 8, | ||||
| @@ -210,12 +212,9 @@ def test_permutation_op(): | |||||
| assert str(output.device) == str(cn) | assert str(output.device) == str(cn) | ||||
| assert output.dtype == dtype | assert output.dtype == dtype | ||||
| # FIXME: remove this sync | |||||
| mge.core.set_option("async_level", 0) | |||||
| test_permutation_op_dtype(np.float32) | test_permutation_op_dtype(np.float32) | ||||
| test_permutation_op_dtype(np.int32) | test_permutation_op_dtype(np.int32) | ||||
| test_permutation_op_dtype(np.int16) | test_permutation_op_dtype(np.int16) | ||||
| mge.core.set_option("async_level", 2) | |||||
| @pytest.mark.skipif( | @pytest.mark.skipif( | ||||
| @@ -517,3 +516,4 @@ def test_rng_empty_tensor(is_symbolic): | |||||
| np.testing.assert_equal(out.numpy().shape, (0,)) | np.testing.assert_equal(out.numpy().shape, (0,)) | ||||
| if is_symbolic is None: | if is_symbolic is None: | ||||
| break | break | ||||
| mge.core.set_option("async_level", 2) | |||||