-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsetup.py
More file actions
30 lines (27 loc) · 1010 Bytes
/
setup.py
File metadata and controls
30 lines (27 loc) · 1010 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import setuptools
with open("README.md",'r',encoding='utf-8') as f:
long_description = f.read()
setuptools.setup(
name="leetDecoding",
version = "0.0.2",
keywords = ["pip", "leetDecoding","LeetDecoding","efficient_linear_decoding"],
description = "Efficient computation library for linear attention.",
long_description=long_description,
long_description_content_type="text/markdown",
license = "MIT Licence",
url = "https://github.com/Computational-Machine-Intelligence/efficient_linear_decoding",
author = "Jiaping Wang",
author_email = "wjp666.s@gmail.com",
packages=setuptools.find_packages(),
# packages=setuptools.find_packages(
# where="efficient_linear_decoding",
# exclude=[
# "tests"
# ]
# ),
platforms = "any",
install_requires=["pycuda", "triton", "pynvml", "pandas", "torch"],
include_package_data=True,
# package_dir={"":"efficient_linear_decoding"},
python_requires=">=3.8",
)