fix sparse_attention imports (#1446)

Commit

4 years ago

fix sparse_attention imports (#1446) Co-authored-by: Jeff Rasley <jerasley@microsoft.com>