liger-kernel-nightly 0.6.2.dev20250905160847__tar.gz → 0.6.2.dev20250913213534__tar.gz

{liger_kernel_nightly-0.6.2.dev20250905160847 → liger_kernel_nightly-0.6.2.dev20250913213534}/.gitignore

@@ -23,4 +23,5 @@ uv.lock
 
 # Benchmark images
 benchmark/visualizations
-.vscode/
+.vscode/
+.coverage

{liger_kernel_nightly-0.6.2.dev20250905160847 → liger_kernel_nightly-0.6.2.dev20250913213534}/Makefile

@@ -5,7 +5,16 @@ all: checkstyle test test-convergence
 
 # Command to run pytest for correctness tests
 test:
-	python -m pytest --disable-warnings test/ --ignore=test/convergence
+	python -m pytest --disable-warnings \
+		-n auto \
+		--dist=load \
+		--cov=src/liger_kernel \
+		--cov-report=term-missing \
+		--ignore=test/convergence \
+		test/
+	coverage combine
+	coverage report -m
+	coverage html
 
 # Command to run ruff for linting and formatting code
 checkstyle:

{liger_kernel_nightly-0.6.2.dev20250905160847 → liger_kernel_nightly-0.6.2.dev20250913213534}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.6.2.dev20250905160847
+Version: 0.6.2.dev20250913213534
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation 
@@ -40,6 +40,8 @@ Requires-Dist: black>=24.4.2; extra == "dev"
 Requires-Dist: isort>=5.13.2; extra == "dev"
 Requires-Dist: pytest>=7.1.2; extra == "dev"
 Requires-Dist: pytest-xdist; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: pytest-rerunfailures; extra == "dev"
 Requires-Dist: datasets>=2.19.2; extra == "dev"
 Requires-Dist: seaborn; extra == "dev"

{liger_kernel_nightly-0.6.2.dev20250905160847 → liger_kernel_nightly-0.6.2.dev20250913213534}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "liger_kernel_nightly"
-version = "0.6.2.dev20250905160847"
+version = "0.6.2.dev20250913213534"
 description = "Efficient Triton kernels for LLM Training"
 urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
 readme = { file = "README.md", content-type = "text/markdown" }
@@ -24,6 +24,36 @@ pythonpath = ["src", "."]
 asyncio_mode = "auto"
 log_cli = true
 log_cli_level = "INFO"
+addopts = [
+    "-n", "auto",
+    "--dist=load",                    # use "load" to distribute tests and let pytest-cov combine coverage
+    "--cov=src/liger_kernel",
+    "--cov-report=term-missing",
+    "--cov-report=html",
+    "--cov-config=pyproject.toml",
+    "--durations=0"
+]
+python_files = "test_*.py"
+testpaths = ["test/"]
+
+[tool.coverage.run]
+branch = true
+parallel = true
+source = ["src/liger_kernel"]
+# xdist uses subprocesses; "multiprocessing" is a safe concurrency choice
+concurrency = ["multiprocessing"]
+
+[tool.coverage.paths]
+liger_kernel = [
+  "src/liger_kernel",
+  "*/site-packages/liger_kernel"
+]
+
+[tool.coverage.report]
+omit = ["test/*"]
+show_missing = true
+skip_covered = false
+
 
 [tool.ruff]
 line-length = 120

{liger_kernel_nightly-0.6.2.dev20250905160847 → liger_kernel_nightly-0.6.2.dev20250913213534}/setup.py

@@ -38,6 +38,8 @@ def get_optional_dependencies():
             "isort>=5.13.2",
             "pytest>=7.1.2",
             "pytest-xdist",
+            "pytest-cov",
+            "pytest-asyncio",
             "pytest-rerunfailures",
             "datasets>=2.19.2",
             "seaborn",

{liger_kernel_nightly-0.6.2.dev20250905160847 → liger_kernel_nightly-0.6.2.dev20250913213534}/src/liger_kernel/transformers/fused_linear_cross_entropy.py

@@ -25,7 +25,8 @@ class LigerFusedLinearCrossEntropyLoss(torch.nn.Module):
         assert reduction in {
             "mean",
             "sum",
-        }, f"reduction must be 'mean' or 'sum'. Got: {reduction}"
+            "none", 
+        }, f"reduction must be 'mean' or 'sum' or 'none'. Got: {reduction}"
         assert softcap is None or softcap > 0, f"softcap must greater than 0.0 or None. Got: {softcap}"
         self.ce_weight = ce_weight
         self.ignore_index = ignore_index

{liger_kernel_nightly-0.6.2.dev20250905160847 → liger_kernel_nightly-0.6.2.dev20250913213534}/src/liger_kernel_nightly.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.6.2.dev20250905160847
+Version: 0.6.2.dev20250913213534
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation 
@@ -40,6 +40,8 @@ Requires-Dist: black>=24.4.2; extra == "dev"
 Requires-Dist: isort>=5.13.2; extra == "dev"
 Requires-Dist: pytest>=7.1.2; extra == "dev"
 Requires-Dist: pytest-xdist; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: pytest-rerunfailures; extra == "dev"
 Requires-Dist: datasets>=2.19.2; extra == "dev"
 Requires-Dist: seaborn; extra == "dev"

{liger_kernel_nightly-0.6.2.dev20250905160847 → liger_kernel_nightly-0.6.2.dev20250913213534}/src/liger_kernel_nightly.egg-info/requires.txt

@@ -9,6 +9,8 @@ black>=24.4.2
 isort>=5.13.2
 pytest>=7.1.2
 pytest-xdist
+pytest-cov
+pytest-asyncio
 pytest-rerunfailures
 datasets>=2.19.2
 seaborn