Skip invalid test cases for LOWER_RIGHT causal variant

voltjia · voltjia · commit 02ceb9803710 · 2025-09-13T10:04:59.000+08:00
diff --git a/tests/test_scaled_dot_product_attention.py b/tests/test_scaled_dot_product_attention.py
@@ -59,6 +59,9 @@ def _generate_random_size():
             atol = 0.025
             rtol = 0.025
 
+        if causal_variant == CausalVariant.LOWER_RIGHT and seq_len_q > seq_len_kv:
+            continue
+
         arguments.append(
             (
                 batch_size,

Original file line number	Diff line number	Diff line change
`@@ -59,6 +59,9 @@ def _generate_random_size():`
`59`	`59`	`atol = 0.025`
`60`	`60`	`rtol = 0.025`
`61`	`61`
	`62`	`+ if causal_variant == CausalVariant.LOWER_RIGHT and seq_len_q > seq_len_kv:`
	`63`	`+ continue`
	`64`	`+`
`62`	`65`	`arguments.append(`
`63`	`66`	`(`
`64`	`67`	`batch_size,`