diff --git a/README.md b/README.md index 8152a8d..20e63a9 100755 --- a/README.md +++ b/README.md @@ -4,6 +4,12 @@ Features: - Support fp32, fp16, bf16. - Kernel size 2, 3, 4. +## Installation + +```bash +pip install causal-conv1d +``` + ## How to use ``` diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..aa04439 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,29 @@ +[project] +name = "causal_conv1d" +readme = "README.md" +dynamic = ["version"] +authors = [ + { name = "Tri Dao", email = "tri@tridao.me" }, +] +requires-python = ">= 3.7" +license = { file = "LICENSE" } # Include a LICENSE file in your repo +dependencies = [ + "torch", + "einops", + "numpy", +] + +[project.optional-dependencies] +dev = [ + "pytest" +] + +[build-system] +requires = [ + "setuptools>=61.0.0", + "wheel", + "torch", + "packaging", + "ninja", +] +build-backend = "setuptools.build_meta" diff --git a/setup.py b/setup.py index b974854..b7454a9 100644 --- a/setup.py +++ b/setup.py @@ -359,10 +359,5 @@ def run(self): else { "bdist_wheel": CachedWheelsCommand, }, - python_requires=">=3.7", - install_requires=[ - "torch", - "packaging", - "ninja", - ], + )