liger-kernel-nightly 0.4.2.dev20241209085735__py3-none-any.whl → 0.4.2.dev20241209195727__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- liger_kernel/env_report.py +10 -1
- {liger_kernel_nightly-0.4.2.dev20241209085735.dist-info → liger_kernel_nightly-0.4.2.dev20241209195727.dist-info}/METADATA +2 -2
- {liger_kernel_nightly-0.4.2.dev20241209085735.dist-info → liger_kernel_nightly-0.4.2.dev20241209195727.dist-info}/RECORD +7 -7
- {liger_kernel_nightly-0.4.2.dev20241209085735.dist-info → liger_kernel_nightly-0.4.2.dev20241209195727.dist-info}/LICENSE +0 -0
- {liger_kernel_nightly-0.4.2.dev20241209085735.dist-info → liger_kernel_nightly-0.4.2.dev20241209195727.dist-info}/NOTICE +0 -0
- {liger_kernel_nightly-0.4.2.dev20241209085735.dist-info → liger_kernel_nightly-0.4.2.dev20241209195727.dist-info}/WHEEL +0 -0
- {liger_kernel_nightly-0.4.2.dev20241209085735.dist-info → liger_kernel_nightly-0.4.2.dev20241209195727.dist-info}/top_level.txt +0 -0
liger_kernel/env_report.py
CHANGED
@@ -24,7 +24,7 @@ def print_env_report():
|
|
24
24
|
cuda_version = (
|
25
25
|
torch.version.cuda if torch.cuda.is_available() else "Not available"
|
26
26
|
)
|
27
|
-
print(f"CUDA version: {cuda_version}")
|
27
|
+
print(f"CUDA version: {cuda_version}")
|
28
28
|
except ImportError:
|
29
29
|
print("PyTorch: Not installed")
|
30
30
|
print("CUDA version: Unable to query")
|
@@ -42,6 +42,15 @@ def print_env_report():
|
|
42
42
|
print(f"Transformers version: {transformers.__version__}")
|
43
43
|
except ImportError:
|
44
44
|
print("Transformers: Not installed")
|
45
|
+
|
46
|
+
try:
|
47
|
+
xpu_version = (
|
48
|
+
torch.version.xpu if torch.xpu.is_available() else "XPU Not Available"
|
49
|
+
)
|
50
|
+
print(f"XPU version: {xpu_version}")
|
51
|
+
except ImportError:
|
52
|
+
print("XPU version: Unable to query")
|
53
|
+
|
45
54
|
|
46
55
|
|
47
56
|
if __name__ == "__main__":
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.1
|
2
2
|
Name: liger_kernel_nightly
|
3
|
-
Version: 0.4.2.
|
3
|
+
Version: 0.4.2.dev20241209195727
|
4
4
|
Summary: Efficient Triton kernels for LLM Training
|
5
5
|
License: BSD 2-CLAUSE LICENSE
|
6
6
|
Copyright 2024 LinkedIn Corporation
|
@@ -99,7 +99,7 @@ Requires-Dist: transformers~=4.0; extra == "transformers"
|
|
99
99
|
</div>
|
100
100
|
<div style="display: block;">
|
101
101
|
<a href="https://github.com/linkedin/Liger-Kernel/actions/workflows/amd-ci.yml">
|
102
|
-
<img src="https://github.com/linkedin/Liger-Kernel/actions/workflows/amd-ci.yml/badge.svg" alt="Build">
|
102
|
+
<img src="https://github.com/linkedin/Liger-Kernel/actions/workflows/amd-ci.yml/badge.svg?event=schedule" alt="Build">
|
103
103
|
</a>
|
104
104
|
</div>
|
105
105
|
</td>
|
@@ -1,5 +1,5 @@
|
|
1
1
|
liger_kernel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
2
|
-
liger_kernel/env_report.py,sha256=
|
2
|
+
liger_kernel/env_report.py,sha256=YHJJCm2y53JWgBUPBcQX5ntUO0_oKv8Q5aHv-kGj_Kw,1385
|
3
3
|
liger_kernel/utils.py,sha256=HJa-xVKOohDn6pLVIx-Fv0V9h0QAL3qZGQNRICI-OpI,249
|
4
4
|
liger_kernel/chunked_loss/__init__.py,sha256=R2wCcz4Y0kTAve926DH3k182XKezpXeACMHj05g9Mm8,346
|
5
5
|
liger_kernel/chunked_loss/cpo_loss.py,sha256=P20txjErLCSfSfToFT8pnuVPqFU4Bbybt3zRXfGEV-0,3122
|
@@ -56,9 +56,9 @@ liger_kernel/transformers/model/qwen2.py,sha256=EyhSSzQOskGjSnCsKMZpd1s5IAIlHd5P
|
|
56
56
|
liger_kernel/transformers/model/qwen2_vl.py,sha256=bIQe2bWiY--G84FhCD29Gdi64_qHP6vbcGsK6vKysQE,8547
|
57
57
|
liger_kernel/triton/__init__.py,sha256=yfRe0zMb47QnqjecZWG7LnanfCTzeku7SgWRAwNVmzU,101
|
58
58
|
liger_kernel/triton/monkey_patch.py,sha256=5BcGKTtdqeYchypBIBopGIWPx1-cFALz7sOKoEsqXJ0,1584
|
59
|
-
liger_kernel_nightly-0.4.2.
|
60
|
-
liger_kernel_nightly-0.4.2.
|
61
|
-
liger_kernel_nightly-0.4.2.
|
62
|
-
liger_kernel_nightly-0.4.2.
|
63
|
-
liger_kernel_nightly-0.4.2.
|
64
|
-
liger_kernel_nightly-0.4.2.
|
59
|
+
liger_kernel_nightly-0.4.2.dev20241209195727.dist-info/LICENSE,sha256=OhzLDHJ0to4a8sodVLELZiCFylZ1NAAYLs-HrjPy0ag,1312
|
60
|
+
liger_kernel_nightly-0.4.2.dev20241209195727.dist-info/METADATA,sha256=HbaHWN8kz7wuxi6HM-zYAyiEMmhdI8Qpf9IsFE6NQSo,22801
|
61
|
+
liger_kernel_nightly-0.4.2.dev20241209195727.dist-info/NOTICE,sha256=njwnoPZLh9AN8SJQzxvCGLHi-8X__AvWRze6joNXIY8,2066
|
62
|
+
liger_kernel_nightly-0.4.2.dev20241209195727.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
|
63
|
+
liger_kernel_nightly-0.4.2.dev20241209195727.dist-info/top_level.txt,sha256=2eghu4hA3LnkM7ElW92tQ8zegWKgSbeo-k-aGe1YnvY,13
|
64
|
+
liger_kernel_nightly-0.4.2.dev20241209195727.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|