diff --git a/test/srt/cpu/test_rope.py b/test/srt/cpu/test_rope.py index 22824e0ca57e..8c1dfe9aa168 100644 --- a/test/srt/cpu/test_rope.py +++ b/test/srt/cpu/test_rope.py @@ -146,12 +146,6 @@ def single_test( (128, 128, 2048, 10000, False, torch.bfloat16, "cpu", 2, 512, 32, 8), (128, 128, 2048, 10000, False, torch.bfloat16, "cpu", 2, 512, 16, 4), (512, 128, 311, 10000, False, torch.bfloat16, "cpu", 3, 39, 4, 2), - (64, 64, 32, 8000, True, torch.float32, "cpu", 32, 32, 1, 1), - (256, 128, 4096, 10000, True, torch.float32, "cpu", 2, 512, 32, 8), - (512, 128, 311, 10000, True, torch.float32, "cpu", 3, 39, 4, 2), - (128, 128, 2048, 10000, False, torch.float32, "cpu", 2, 512, 32, 8), - (128, 128, 2048, 10000, False, torch.float32, "cpu", 2, 512, 16, 4), - (512, 128, 311, 10000, False, torch.float32, "cpu", 3, 39, 4, 2), ] for (