We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 79d177d commit dc333edCopy full SHA for dc333ed
fbgemm_gpu/experimental/gen_ai/test/attention/blackwell_fmha_test.py
@@ -15,18 +15,11 @@
15
from fbgemm_gpu.experimental.gen_ai.attention.cutlass_blackwell_fmha import (
16
cutlass_blackwell_fmha_func,
17
)
18
-from hypothesis import HealthCheck, Verbosity
19
from parameterized import parameterized
20
21
from .attention_ref_fp8 import attention_ref_fp8
22
from .test_utils import attention_ref, generate_qkv, generate_random_padding_mask
23
24
-common_settings = {
25
- "verbosity": Verbosity.normal,
26
- "max_examples": 200,
27
- "deadline": None,
28
- "suppress_health_check": [HealthCheck.filter_too_much, HealthCheck.data_too_large],
29
-}
30
31
DEBUG = False
32
SEED = 2
0 commit comments