diff --git a/tests/st/test_custom_grid_sample.py b/tests/st/test_custom_grid_sample.py index 76887a27de0214cd04c7c4874b88ea8737079d10..075338ff3d766ae5bb0b3c87ab224d9a8817e2ab 100644 --- a/tests/st/test_custom_grid_sample.py +++ b/tests/st/test_custom_grid_sample.py @@ -142,10 +142,10 @@ def run_grid_sample(net, exec_mode, input_dtype, grid_dtype, align_corners, padd @pytest.mark.parametrize("padding_mode", ["border"]) @pytest.mark.parametrize("interpolation_mode", ["bilinear"]) @pytest.mark.parametrize("n", [1]) -@pytest.mark.parametrize("c", [1536]) +@pytest.mark.parametrize("c", [1536, 8]) @pytest.mark.parametrize("h_in,w_in", [(24, 24)]) @pytest.mark.parametrize("h_out,w_out", [(1024, 1)]) -def test_rope_v3_interleave(exec_mode, input_dtype, grid_dtype, align_corners, padding_mode, interpolation_mode, n, c, h_in, w_in, h_out, w_out): +def test_grid_sample(exec_mode, input_dtype, grid_dtype, align_corners, padding_mode, interpolation_mode, n, c, h_in, w_in, h_out, w_out): """ Feature:aclnnGridSample kernel. Description: test for GridSampleExt ops.