Skip to content

Commit

Permalink
unskip test_unsupported_alignment
Browse files Browse the repository at this point in the history
  • Loading branch information
tenpercent committed Feb 28, 2024
1 parent 6fbb383 commit 1db3a5a
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions tests/test_mem_eff_attention.py
Original file line number Diff line number Diff line change
Expand Up @@ -1560,9 +1560,6 @@ def test_unsupported_stride_lastdim(op: Type[fmha.AttentionFwOpBase]):
0, 3, 1, 2
)

if skip_reasons := op.not_supported_reasons(fmha.Inputs(q, q, q)):
pytest.skip("; ".join(skip_reasons))

try:
fmha.memory_efficient_attention(q, q, q, op=(op, None))
except ValueError as e:
Expand All @@ -1579,9 +1576,6 @@ def test_unsupported_stride_lastdim(op: Type[fmha.AttentionFwOpBase]):
def test_unsupported_stride_alignment(op: Type[fmha.AttentionFwOpBase]):
q = torch.empty([1, 2, 1, 33], device="cuda", dtype=torch.float16)[:, :, :, :32]

if skip_reasons := op.not_supported_reasons(fmha.Inputs(q, q, q)):
pytest.skip("; ".join(skip_reasons))

try:
fmha.memory_efficient_attention(q, q, q, op=(op, None))
except ValueError as e:
Expand Down

0 comments on commit 1db3a5a

Please sign in to comment.