mirror of
https://github.com/linkedin/Liger-Kernel.git
synced 2026-10-12 02:17:47 +00:00
Andre Slavescu
d338f4b992
## 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
39 lines
800 B
TOML
39 lines
800 B
TOML
[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"
|