Merge pull request #193 from anthonyhu/pyproject-build

Use pyproject.toml to specify build dependencies
This commit is contained in:
Tri Dao 2023-05-11 21:26:28 -04:00 committed by GitHub
commit 36d0a19f1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 1 deletions

View File

@ -51,7 +51,7 @@ pip install flash-attn
Alternatively you can compile from source:
```
python setup.py install
python -m pip install .
```
Interface: `src/flash_attention.py`

3
pyproject.toml Normal file
View File

@ -0,0 +1,3 @@
[build-system]
requires = ["packaging", "setuptools", "torch", "wheel"]
build-backend = "setuptools.build_meta"