liger-kernel-nightly 0.4.2.dev20241119054537__tar.gz → 0.4.2.dev20241119054729__tar.gz

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.
Files changed (63) hide show
  1. {liger_kernel_nightly-0.4.2.dev20241119054537/src/liger_kernel_nightly.egg-info → liger_kernel_nightly-0.4.2.dev20241119054729}/PKG-INFO +1 -1
  2. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/pyproject.toml +1 -1
  3. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/rms_norm.py +5 -1
  4. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729/src/liger_kernel_nightly.egg-info}/PKG-INFO +1 -1
  5. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/LICENSE +0 -0
  6. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/NOTICE +0 -0
  7. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/README.md +0 -0
  8. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/setup.cfg +0 -0
  9. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/chunked_loss/__init__.py +0 -0
  10. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/chunked_loss/cpo_loss.py +0 -0
  11. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/chunked_loss/dpo_loss.py +0 -0
  12. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/chunked_loss/fused_linear_preference.py +0 -0
  13. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/chunked_loss/orpo_loss.py +0 -0
  14. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/env_report.py +0 -0
  15. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/__init__.py +0 -0
  16. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/cross_entropy.py +0 -0
  17. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/experimental/embedding.py +0 -0
  18. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/experimental/mm_int8int2.py +0 -0
  19. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/fused_linear_cross_entropy.py +0 -0
  20. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/fused_linear_jsd.py +0 -0
  21. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/geglu.py +0 -0
  22. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/group_norm.py +0 -0
  23. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/jsd.py +0 -0
  24. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/kl_div.py +0 -0
  25. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/layer_norm.py +0 -0
  26. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/qwen2vl_mrope.py +0 -0
  27. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/rope.py +0 -0
  28. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/swiglu.py +0 -0
  29. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/utils.py +0 -0
  30. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/__init__.py +0 -0
  31. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/auto_model.py +0 -0
  32. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/cross_entropy.py +0 -0
  33. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/experimental/embedding.py +0 -0
  34. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/functional.py +0 -0
  35. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/fused_linear_cross_entropy.py +0 -0
  36. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/fused_linear_jsd.py +0 -0
  37. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/geglu.py +0 -0
  38. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/group_norm.py +0 -0
  39. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/jsd.py +0 -0
  40. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/kl_div.py +0 -0
  41. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/layer_norm.py +0 -0
  42. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/__init__.py +0 -0
  43. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/gemma.py +0 -0
  44. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/gemma2.py +0 -0
  45. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/llama.py +0 -0
  46. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/mistral.py +0 -0
  47. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/mixtral.py +0 -0
  48. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/mllama.py +0 -0
  49. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/phi3.py +0 -0
  50. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/qwen2.py +0 -0
  51. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/model/qwen2_vl.py +0 -0
  52. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/monkey_patch.py +0 -0
  53. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/qwen2vl_mrope.py +0 -0
  54. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/rms_norm.py +0 -0
  55. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/rope.py +0 -0
  56. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/swiglu.py +0 -0
  57. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/transformers/trainer_integration.py +0 -0
  58. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/triton/__init__.py +0 -0
  59. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/triton/monkey_patch.py +0 -0
  60. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel_nightly.egg-info/SOURCES.txt +0 -0
  61. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel_nightly.egg-info/dependency_links.txt +0 -0
  62. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel_nightly.egg-info/requires.txt +0 -0
  63. {liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel_nightly.egg-info/top_level.txt +0 -0
{liger_kernel_nightly-0.4.2.dev20241119054537/src/liger_kernel_nightly.egg-info → liger_kernel_nightly-0.4.2.dev20241119054729}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: liger_kernel_nightly
3
- Version: 0.4.2.dev20241119054537
3
+ Version: 0.4.2.dev20241119054729
4
4
  Summary: Efficient Triton kernels for LLM Training
5
5
  License: BSD 2-CLAUSE LICENSE
6
6
  Copyright 2024 LinkedIn Corporation
{liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "liger_kernel_nightly"
7
- version = "0.4.2.dev20241119054537"
7
+ version = "0.4.2.dev20241119054729"
8
8
  description = "Efficient Triton kernels for LLM Training"
9
9
  urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
10
10
  readme = { file = "README.md", content-type = "text/markdown" }
{liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729}/src/liger_kernel/ops/rms_norm.py RENAMED
@@ -264,7 +264,11 @@ def rms_norm_backward(
264
264
  dY = dY.view(-1, dim)
265
265
  n_rows, n_cols = dY.shape
266
266
 
267
- sm_count = torch.cuda.get_device_properties(X.device).multi_processor_count
267
+ if X.device.type == "cuda":
268
+ sm_count = torch.cuda.get_device_properties(X.device).multi_processor_count
269
+ elif X.device.type == "xpu":
270
+ sm_count = torch.xpu.get_device_properties(X.device).gpu_subslice_count
271
+
268
272
  # fp32 for numerical stability especially.
269
273
  _dW = torch.empty((sm_count, n_cols), dtype=torch.float32, device=W.device)
270
274
 
{liger_kernel_nightly-0.4.2.dev20241119054537 → liger_kernel_nightly-0.4.2.dev20241119054729/src/liger_kernel_nightly.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: liger_kernel_nightly
3
- Version: 0.4.2.dev20241119054537
3
+ Version: 0.4.2.dev20241119054729
4
4
  Summary: Efficient Triton kernels for LLM Training
5
5
  License: BSD 2-CLAUSE LICENSE
6
6
  Copyright 2024 LinkedIn Corporation