allow for uploading to pypi

BioGeek · Nov 15, 2022 · b0eac32 · b0eac32
1 parent 2e33fc8
commit b0eac32
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,21 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+
+# C extensions
+*.so
+
+# Distribution / packaging
+bin/
+build/
+develop-eggs/
+dist/
+eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1,7 @@
+recursive-include csrc *.cu
+recursive-include csrc *.h
+recursive-include csrc *.cpp
+
+recursive-include flash_attn *.cu
+recursive-include flash_attn *.h
+recursive-include flash_attn *.cpp
diff --git a/Makefile b/Makefile
@@ -0,0 +1,9 @@
+
+clean_dist:
+	rm -rf dist/*
+
+create_dist: clean_dist
+	python setup.py sdist
+
+upload_package: create_dist
+	twine upload dist/*
diff --git a/setup.py b/setup.py
@@ -164,8 +164,8 @@ def append_nvcc_threads(nvcc_extra_args):
     url="https://github.com/HazyResearch/flash-attention",
     classifiers=[
         "Programming Language :: Python :: 3",
-        "License :: Apache 2.0",
-        "Operating System :: Linux",
+        "License :: OSI Approved :: Apache Software License",
+        "Operating System :: Unix",
     ],
     ext_modules=ext_modules,
     cmdclass={"build_ext": BuildExtension} if ext_modules else {},