Fixed tests

Author: mseeger

tests/test_model.py

@@ -1471,6 +1471,8 @@ def assert_sdpa_backend(original_fn, query, k_and_v, mask, return_scores):
         elif expected is SDPBackend.EFFICIENT_ATTENTION:
             assert mem_efficient_sdp_enabled(), "mem_efficient_sdp_enabled() is False"
             if (not enable_gqa) or mask is None:
+                # At present, `SDPBackend.EFFICIENT_ATTENTION` does not support
+                # `enabla_gqa=True` and a mask specified
                 assert can_use_efficient_attention(params, True), "can_use_efficient_attention(params, True) is False"
         elif expected is SDPBackend.MATH:
             assert math_sdp_enabled(), "math_sdp_enabled() is False"
@@ -1540,6 +1542,8 @@ def assert_sdpa_backend(original_fn, query, k_and_v, mask, return_scores):
         elif expected is SDPBackend.EFFICIENT_ATTENTION:
             assert mem_efficient_sdp_enabled(), "mem_efficient_sdp_enabled() is False"
             if (not enable_gqa) or mask is None:
+                # At present, `SDPBackend.EFFICIENT_ATTENTION` does not support
+                # `enabla_gqa=True` and a mask specified
                 assert can_use_efficient_attention(params, True), "can_use_efficient_attention(params, True) is False"
         elif expected is SDPBackend.MATH:
             assert math_sdp_enabled(), "math_sdp_enabled() is False"