Skip to content

GitLab

Unverified Commit 006ead9d authored by Brayden Zhong's avatar Brayden Zhong Committed by GitHub
Browse files

[FA][Test] Fix Sparse FA test (#6306)

parent 0d503090
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
sgl-kernel/tests/test_sparse_flash_attn.py
View file @ 006ead9d
......@@ -10,6 +10,7 @@ from sgl_kernel.sparse_flash_attn import (
sparse_attn_func,
sparse_attn_varlen_func,
)
from test_flash_attention import construct_local_mask
def ref_attn(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment