Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/fastmoe.git" did not exist on "4682c1d007df1dc41ddef7bc33c7a7ccb8031a09"
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 ( ...@@ -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