liger-kernel-nightly 0.5.0.dev20241210031928__tar.gz → 0.5.0.dev20241210032000__tar.gz

{liger_kernel_nightly-0.5.0.dev20241210031928/src/liger_kernel_nightly.egg-info → liger_kernel_nightly-0.5.0.dev20241210032000}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.5.0.dev20241210031928
+Version: 0.5.0.dev20241210032000
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation 

{liger_kernel_nightly-0.5.0.dev20241210031928 → liger_kernel_nightly-0.5.0.dev20241210032000}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "liger_kernel_nightly"
-version = "0.5.0.dev20241210031928"
+version = "0.5.0.dev20241210032000"
 description = "Efficient Triton kernels for LLM Training"
 urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
 readme = { file = "README.md", content-type = "text/markdown" }

{liger_kernel_nightly-0.5.0.dev20241210031928 → liger_kernel_nightly-0.5.0.dev20241210032000}/src/liger_kernel/env_report.py

@@ -1,5 +1,6 @@
 import platform
 import sys
+from importlib.metadata import version
 
 
 def print_env_report():
@@ -17,6 +18,11 @@ def print_env_report():
     print(f"Operating System: {platform.platform()}")
     print(f"Python version: {sys.version.split()[0]}")
 
+    try:
+        print(f"Liger Kernel version: {version('liger-kernel')}")
+    except ImportError:
+        print("Liger Kernel: Not installed")
+
     try:
         import torch
 
@@ -25,9 +31,17 @@ def print_env_report():
             torch.version.cuda if torch.cuda.is_available() else "Not available"
         )
         print(f"CUDA version: {cuda_version}")
+        hip_version = (
+            torch.version.hip
+            if torch.cuda.is_available() and torch.version.hip
+            else "Not available"
+        )
+        print(f"HIP(ROCm) version: {hip_version}")
+
     except ImportError:
         print("PyTorch: Not installed")
         print("CUDA version: Unable to query")
+        print("HIP(ROCm) version: Unable to query")
 
     try:
         import triton

{liger_kernel_nightly-0.5.0.dev20241210031928 → liger_kernel_nightly-0.5.0.dev20241210032000/src/liger_kernel_nightly.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: liger_kernel_nightly
-Version: 0.5.0.dev20241210031928
+Version: 0.5.0.dev20241210032000
 Summary: Efficient Triton kernels for LLM Training
 License: BSD 2-CLAUSE LICENSE
         Copyright 2024 LinkedIn Corporation