Skip to content

GitLab

Unverified Commit 3c4053b7 authored by janEbert's avatar janEbert Committed by GitHub
Browse files

Make FA3 externally importable (#1053) 

Library name to import is `flash_attn_interface`, which matches the
test.
parent d8f104e9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
hopper/setup.py
View file @ 3c4053b7
...@@ -268,6 +268,7 @@ setup( ...@@ -268,6 +268,7 @@ setup(
"benchmarks", "benchmarks",
) )
), ),
py_modules=["flash_attn_interface"],
description="FlashAttention-3", description="FlashAttention-3",
long_description=long_description, long_description=long_description,
long_description_content_type="text/markdown", long_description_content_type="text/markdown",
......
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