Skip to content

GitLab

"examples/pytorch/vscode:/vscode.git/clone" did not exist on "d4d4fa1eaf86b31821cf8c14d3a1cc4a358f3b6e"
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
Show 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 ( ...@@ -10,6 +10,7 @@ from sgl_kernel.sparse_flash_attn import (
sparse_attn_func, sparse_attn_func,
sparse_attn_varlen_func, sparse_attn_varlen_func,
) )
from test_flash_attention import construct_local_mask
def ref_attn( 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