diff --git a/test/registered/dp_attn/test_dp_attention.py b/test/registered/dp_attn/test_dp_attention.py index 0bc16af04..95d34e76b 100644 --- a/test/registered/dp_attn/test_dp_attention.py +++ b/test/registered/dp_attn/test_dp_attention.py @@ -56,9 +56,6 @@ class TestDPAttentionDP2TP2( "--enable-dp-attention", "--dp", "2", - "--enable-torch-compile", - "--torch-compile-max-bs", - "2", ], )