mirror of
https://github.com/linkedin/Liger-Kernel.git
synced 2026-10-12 02:17:47 +00:00
[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:
@@ -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"
|
||||
@@ -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",
|
||||
]
|
||||
},
|
||||
)
|
||||
Reference in New Issue
Block a user