[BUILD] Add pyproject.toml (#150)

## Summary
<!--- This is a required section; please describe the main purpose of
this proposed code change. --->
added pyproject.toml

<!---
## Details
This is an optional section; is there anything specific that reviewers
should be aware of?
--->

## Testing Done
<!--- This is a required section; please describe how this change was
tested. --->
ran `pip install -e .` and it built successfully

<!-- 
Replace BLANK with your device type. For example, A100-80G-PCIe

Complete the following tasks before sending your PR, and replace `[ ]`
with
`[x]` to indicate you have done them. 
-->

- Hardware Type: RTX 3090
- [x] run `make test` to ensure correctness
- [x] run `make checkstyle` to ensure code style
- [x] run `make test-convergence` to ensure convergence
This commit is contained in:
Andre Slavescu
2024-09-03 13:47:57 -07:00
committed by GitHub
parent 3c71ea4ffa
commit d338f4b992
2 changed files with 38 additions and 45 deletions
+38
View File
@@ -0,0 +1,38 @@
[build-system]
requires = ["setuptools>=42", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "liger_kernel"
version = "0.2.1"
description = "Efficient Triton kernels for LLM Training"
urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
readme = { file = "README.md", content-type = "text/markdown" }
license = { file = "LICENSE" }
dependencies = [
"torch>=2.1.2",
"triton>=2.3.0",
"transformers>=4.42.0"
]
[project.optional-dependencies]
dev = [
"matplotlib>=3.7.2",
"flake8>=4.0.1.1",
"black>=24.4.2",
"isort>=5.13.2",
"pytest>=7.1.2",
"datasets>=2.19.2"
]
[tool.setuptools.packages.find]
include = ["src"]
[tool.pytest.ini_options]
pythonpath = [
"src",
"."
]
asyncio_mode = "auto"
log_cli = true
log_cli_level = "INFO"
-45
View File
@@ -1,45 +0,0 @@
from setuptools import find_namespace_packages, setup
__version__ = "0.2.1"
setup(
name="liger_kernel",
version=__version__,
description="Efficient Triton kernels for LLM Training",
long_description=open("README.md").read(),
long_description_content_type="text/markdown",
license="BSD-2-Clause",
url="https://github.com/linkedin/Liger-Kernel",
package_dir={"": "src"},
packages=find_namespace_packages(where="src"),
classifiers=[
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"Intended Audience :: Education",
"License :: OSI Approved :: BSD License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Topic :: Software Development :: Libraries",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
keywords="triton,kernels,LLM training,deep learning,Hugging Face,PyTorch,GPU optimization",
include_package_data=True,
install_requires=[
"torch>=2.1.2",
"triton>=2.3.0",
"transformers>=4.42.0",
],
extras_require={
"dev": [
"matplotlib>=3.7.2",
"flake8>=4.0.1.1",
"black>=24.4.2",
"isort>=5.13.2",
"pytest>=7.1.2",
"datasets>=2.19.2",
]
},
)