liger-kernel-nightly 0.4.2.dev20241209060920__tar.gz → 0.4.2.dev20241209085735__tar.gz

{liger_kernel_nightly-0.4.2.dev20241209060920/src/liger_kernel_nightly.egg-info → liger_kernel_nightly-0.4.2.dev20241209085735}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.4.2.dev20241209060920
+Version: 0.4.2.dev20241209085735
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation 
@@ -42,9 +42,16 @@ Requires-Dist: flake8>=4.0.1.1; extra == "dev"
 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-rerunfailures; extra == "dev"
 Requires-Dist: datasets>=2.19.2; extra == "dev"
 Requires-Dist: torchvision>=0.16.2; extra == "dev"
 Requires-Dist: seaborn; extra == "dev"
+Provides-Extra: amd
+Requires-Dist: torch>=2.6.0.dev; extra == "amd"
+Requires-Dist: setuptools-scm>=8; extra == "amd"
+Requires-Dist: torchvision>=0.20.0.dev; extra == "amd"
+Requires-Dist: triton>=3.0.0; extra == "amd"
 
 <a name="readme-top"></a>
 
@@ -191,6 +198,8 @@ To install from source:
 git clone https://github.com/linkedin/Liger-Kernel.git
 cd Liger-Kernel
 pip install -e .
+# or if installing on amd platform
+pip install -e .[amd] --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2 # rocm6.2
 # or if using transformers
 pip install -e .[transformers]
 ```

{liger_kernel_nightly-0.4.2.dev20241209060920 → liger_kernel_nightly-0.4.2.dev20241209085735}/README.md

@@ -143,6 +143,8 @@ To install from source:
 git clone https://github.com/linkedin/Liger-Kernel.git
 cd Liger-Kernel
 pip install -e .
+# or if installing on amd platform
+pip install -e .[amd] --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2 # rocm6.2
 # or if using transformers
 pip install -e .[transformers]
 ```

{liger_kernel_nightly-0.4.2.dev20241209060920 → liger_kernel_nightly-0.4.2.dev20241209085735}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "liger_kernel_nightly"
-version = "0.4.2.dev20241209060920"
+version = "0.4.2.dev20241209085735"
 description = "Efficient Triton kernels for LLM Training"
 urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
 readme = { file = "README.md", content-type = "text/markdown" }
@@ -14,6 +14,7 @@ dependencies = [
     "triton>=2.3.1",
 ]
 
+
 [project.optional-dependencies]
 transformers = [
     "transformers~=4.0"
@@ -27,11 +28,22 @@ dev = [
     "black>=24.4.2",
     "isort>=5.13.2",
     "pytest>=7.1.2",
+    "pytest-xdist", 
+    "pytest-rerunfailures",
     "datasets>=2.19.2",
     "torchvision>=0.16.2",
     "seaborn",
 ]
 
+amd = [
+    "torch>=2.6.0.dev",
+    "setuptools-scm>=8",
+    "torchvision>=0.20.0.dev",
+    "triton>=3.0.0",
+]
+
+
+
 [tool.setuptools.packages.find]
 where = ["src"]
 include = ["liger_kernel", "liger_kernel.*"]

{liger_kernel_nightly-0.4.2.dev20241209060920 → liger_kernel_nightly-0.4.2.dev20241209085735/src/liger_kernel_nightly.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.4.2.dev20241209060920
+Version: 0.4.2.dev20241209085735
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation 
@@ -42,9 +42,16 @@ Requires-Dist: flake8>=4.0.1.1; extra == "dev"
 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-rerunfailures; extra == "dev"
 Requires-Dist: datasets>=2.19.2; extra == "dev"
 Requires-Dist: torchvision>=0.16.2; extra == "dev"
 Requires-Dist: seaborn; extra == "dev"
+Provides-Extra: amd
+Requires-Dist: torch>=2.6.0.dev; extra == "amd"
+Requires-Dist: setuptools-scm>=8; extra == "amd"
+Requires-Dist: torchvision>=0.20.0.dev; extra == "amd"
+Requires-Dist: triton>=3.0.0; extra == "amd"
 
 <a name="readme-top"></a>
 
@@ -191,6 +198,8 @@ To install from source:
 git clone https://github.com/linkedin/Liger-Kernel.git
 cd Liger-Kernel
 pip install -e .
+# or if installing on amd platform
+pip install -e .[amd] --extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2 # rocm6.2
 # or if using transformers
 pip install -e .[transformers]
 ```

{liger_kernel_nightly-0.4.2.dev20241209060920 → liger_kernel_nightly-0.4.2.dev20241209085735}/src/liger_kernel_nightly.egg-info/requires.txt

@@ -1,6 +1,12 @@
 torch>=2.1.2
 triton>=2.3.1
 
+[amd]
+torch>=2.6.0.dev
+setuptools-scm>=8
+torchvision>=0.20.0.dev
+triton>=3.0.0
+
 [dev]
 transformers>=4.44.2
 trl>=0.11.0
@@ -9,6 +15,8 @@ flake8>=4.0.1.1
 black>=24.4.2
 isort>=5.13.2
 pytest>=7.1.2
+pytest-xdist
+pytest-rerunfailures
 datasets>=2.19.2
 torchvision>=0.16.2
 seaborn